Spaces:

tbitai
/

bayes-or-spam

Running

tbitai commited on Oct 3, 2024

Commit

908a1af

verified ·

1 Parent(s): f12cfe9

Fix type of LLM prediction

Files changed (1) hide show

app.py CHANGED Viewed

@@ -68,7 +68,7 @@ def predict_nn(text):
 def predict_llm(text):
     embedding = st_model.encode(text)
-    return llm_model(np.array([embedding]))[0][0].numpy()
 MODELS = [
     BAYES := "Bayes Enron1 spam",

 def predict_llm(text):
     embedding = st_model.encode(text)
+    return float(llm_model(np.array([embedding]))[0][0].numpy())
 MODELS = [
     BAYES := "Bayes Enron1 spam",