Spaces:
Sleeping
Sleeping
update
Browse files
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.
|
54 |
-
AutoModel.from_pretrained(config.db_config.
|
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")
|