Spaces:

tbitai
/

bayes-or-spam

Sleeping

tbitai commited on Oct 3, 2024

Commit

c078837

verified ·

1 Parent(s): 908a1af

Fix type of NN prediction

Files changed (1) hide show

app.py CHANGED Viewed

@@ -64,7 +64,7 @@ def predict_bayes(text, intr_threshold, unbiased=False):
     return combine(interesting_probs)
 def predict_nn(text):
-    return nn_model(np.array([text]))[0][0].numpy()
 def predict_llm(text):
     embedding = st_model.encode(text)

     return combine(interesting_probs)
 def predict_nn(text):
+    return float(nn_model(np.array([text]))[0][0].numpy())
 def predict_llm(text):
     embedding = st_model.encode(text)