Eklavya73 commited on
Commit
e7ec69a
Β·
verified Β·
1 Parent(s): 1403448

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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("all-mpnet-base-v2")
 
 
 
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")