Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,10 @@ LOG_PATH = os.path.join(tempfile.gettempdir(), "routing_evaluation_log.csv")
|
|
| 32 |
|
| 33 |
# ββ Load Models ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 34 |
print("Loading SBERT model...")
|
| 35 |
-
sbert = SentenceTransformer(
|
|
|
|
|
|
|
|
|
|
| 36 |
|
| 37 |
print("Loading classifiers...")
|
| 38 |
tag_model = joblib.load(MODEL_DIR / "sbert_classifier.pkl")
|
|
|
|
| 32 |
|
| 33 |
# ββ Load Models ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 34 |
print("Loading SBERT model...")
|
| 35 |
+
sbert = SentenceTransformer(
|
| 36 |
+
"sentence-transformers/all-mpnet-base-v2",
|
| 37 |
+
cache_folder="/data"
|
| 38 |
+
)
|
| 39 |
|
| 40 |
print("Loading classifiers...")
|
| 41 |
tag_model = joblib.load(MODEL_DIR / "sbert_classifier.pkl")
|