muryshev commited on
Commit
4daae4e
·
1 Parent(s): 1c50aa4
Files changed (1) hide show
  1. main.py +7 -0
main.py CHANGED
@@ -20,6 +20,7 @@ from routes.llm import router as llm_router
20
  from routes.llm_config import router as llm_config_router
21
  from routes.llm_prompt import router as llm_prompt_router
22
  from common.common import configure_logging
 
23
 
24
  # Загружаем переменные из .env
25
  dotenv.load_dotenv()
@@ -42,6 +43,12 @@ configure_logging(
42
  )
43
 
44
 
 
 
 
 
 
 
45
 
46
  app = FastAPI(title="Assistant control panel")
47
 
 
20
  from routes.llm_config import router as llm_config_router
21
  from routes.llm_prompt import router as llm_prompt_router
22
  from common.common import configure_logging
23
+ from transformers import AutoTokenizer, AutoModel
24
 
25
  # Загружаем переменные из .env
26
  dotenv.load_dotenv()
 
43
  )
44
 
45
 
46
+ print("Downloading model to cache...")
47
+ AutoTokenizer.from_pretrained(config.db_config.faiss.model_embedding_path)
48
+ AutoModel.from_pretrained(config.db_config.faiss.model_embedding_path)
49
+
50
+ print("Model cached successfully.")
51
+
52
 
53
  app = FastAPI(title="Assistant control panel")
54