Spaces:

NhatHuy1110
/

ArXiv_Abstract_Classifier

Runtime error

NhatHuy1110 commited on Oct 24, 2025

Commit

0420d4e

verified ·

1 Parent(s): 9554032

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ EMB_MODEL_NAME = (ART/"emb_model_name.txt").read_text().strip()
 @st.cache_resource(show_spinner=False)
 def load_models():
-    emb = SentenceTransformer(EMB_MODEL_NAME)
     clf = joblib.load(ART/"lgbm_model.pkl")
     nn = joblib.load(ART/"nn_index.pkl")
     tfidf = joblib.load(ART/"tfidf_explainer.pkl")

 @st.cache_resource(show_spinner=False)
 def load_models():
+    emb = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
     clf = joblib.load(ART/"lgbm_model.pkl")
     nn = joblib.load(ART/"nn_index.pkl")
     tfidf = joblib.load(ART/"tfidf_explainer.pkl")