main not PR

diff --git a/src/main.rs b/src/main.rs
index 4a11d2bfdf97809afbcbb11eb2c575c24b25d10a..24ccbfbd04f0ead2233bab0a5681d435afa162ee 100644 (file)
--- a/src/main.rs
+++ b/src/main.rs
@@ -92,7 +92,6 @@
 //! 
 //! ## TODO
 //! 
-//! - why model from PR not main?
 //! - rename cv-*
 //! 
 use candle_transformers::models::bert::{BertModel, Config, HiddenAct, DTYPE};
@@ -201,7 +200,7 @@ struct Args {
 fn build_model_and_tokenizer(args: &Args) -> Result<(BertModel, Tokenizer)> {
     let device = candle_core::Device::Cpu;
     let default_model = "sentence-transformers/all-MiniLM-L6-v2".to_string();
-    let default_revision = "refs/pr/21".to_string();
+    let default_revision = "main".to_string();
     let (model_id, revision) = match (args.model_id.to_owned(), args.revision.to_owned()) {
         (Some(model_id), Some(revision)) => (model_id, revision),
         (Some(model_id), None) => (model_id, "main".to_string()),