muryshev commited on
Commit
5dee1a1
·
1 Parent(s): 744a170
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -50,8 +50,8 @@ tmp_path = Path(os.environ.get("APP_TMP_PATH", '.')) / 'tmp.json'
50
  tmp_path.unlink(missing_ok=True)
51
 
52
  print("Downloading model to cache...")
53
- AutoTokenizer.from_pretrained(config.db_config.faiss.model_embedding_path)
54
- AutoModel.from_pretrained(config.db_config.faiss.model_embedding_path)
55
  print("Model cached successfully.")
56
 
57
  app = FastAPI(title="Assistant control panel")
 
50
  tmp_path.unlink(missing_ok=True)
51
 
52
  print("Downloading model to cache...")
53
+ AutoTokenizer.from_pretrained(config.db_config.search.vectorizer_path)
54
+ AutoModel.from_pretrained(config.db_config.search.vectorizer_path)
55
  print("Model cached successfully.")
56
 
57
  app = FastAPI(title="Assistant control panel")