query_opt

diff --git a/src/main.rs b/src/main.rs
index 0c2f96d0db320793a11482a96546a4cd808cc3fe..16aceda83a8675e9573ee38c738eb6b9b1e742d6 100644 (file)
--- a/src/main.rs
+++ b/src/main.rs
@@ -85,7 +85,7 @@
 //! 
 //! - why model from PR not main?
 //! - init-model command
-//! 
+//! - rename cv-*
 //! 
 use candle_transformers::models::bert::{BertModel, Config, HiddenAct, DTYPE};
 
@@ -239,13 +239,14 @@ fn init_database(dbname: String, dbhost: String, dbuser: String, dbpassword: Str
        .connect(NoTls)?;
 
     let result =
-       client.query("SELECT 1 FROM pg_database WHERE datname=$1",&[&dbname])?;
-
-    if result.is_empty() {
-       println!("creating database {}", dbname);
-       client.execute(&format!("create database {}",dbname),&[])?;
-    } else {
-       println!("database {} exists already", dbname);
+       client.query_opt("SELECT 1 FROM pg_database WHERE datname=$1",&[&dbname])?;
+
+    match result {
+       Some(_) => {
+           println!("creating database {}", dbname);
+           client.execute(&format!("create database {}",dbname),&[])?;
+       },
+       None => println!("database {} exists already", dbname),
     }
     let _ = client.close();
 
@@ -278,6 +279,7 @@ fn get_embeddings(input: &String, model: &BertModel, mut tokenizer: Tokenizer) -
         .map_err(E::msg)?
         .get_ids()
         .to_vec();
+    println!("tokens {:?}",tokens);
     let token_ids = Tensor::new(&tokens[..], device)?.unsqueeze(0)?;
     let token_type_ids = token_ids.zeros_like()?;
     println!("Loaded and encoded {:?}", start.elapsed());