Spaces:
Sleeping
Sleeping
update
Browse files
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 |
|