Update utils/predict.py

utils/predict.py

@@ -1,24 +1,17 @@
-from pathlib import Path
 import joblib
 from sentence_transformers import SentenceTransformer
 
-BASE_DIR = Path(__file__).resolve().parents[1]
-MODEL_DIR = BASE_DIR / "model"
-
 def load_model():
-    logistic_model = joblib.load(MODEL_DIR / "logistic_model.pkl")
+    clf = joblib.load("model/logistic_model.pkl")
 
     s2v_model = SentenceTransformer(
-        "Pachinee/sentence2vec-brd"
+        "Pachinee/sentence2vec-brd"  # ← Hugging Face Model
     )
 
-    return logistic_model, s2v_model
+    return clf, s2v_model
 
 
-def predict_label(texts, logistic_model, s2v_model):
-    embeddings = s2v_model.encode(
-        list(texts),
-        convert_to_numpy=True
-    )
-    preds = logistic_model.predict(embeddings)
+def predict_label(texts, clf, s2v_model):
+    embeddings = s2v_model.encode(list(texts))
+    preds = clf.predict(embeddings)
     return ["Clear" if p == 1 else "Unclear" for p in preds]