Spaces:
No application file
No application file
File size: 458 Bytes
109d1d4 da10199 109d1d4 da10199 109d1d4 da10199 109d1d4 da10199 109d1d4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import joblib
from sentence_transformers import SentenceTransformer
def load_model():
clf = joblib.load("model/logistic_model.pkl")
s2v_model = SentenceTransformer(
"Pachinee/sentence2vec-brd" # ← Hugging Face Model
)
return clf, s2v_model
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]
|