Spaces:

SerialGuy
/

ai-vs-human

Running

SerialGuy commited on Apr 18, 2025

Commit

3b9969b

1 Parent(s): 2100669

Fixing bugs

Files changed (1) hide show

predictor.py CHANGED Viewed

@@ -44,5 +44,5 @@ def preprocess_text(text):
 def predict_text(text):
     X_sample = preprocess_text(text)
     prediction = model.predict(X_sample)[0]
-    confidence = model.predict_proba(X_sample)[0][:, 1]
     return prediction, confidence

 def predict_text(text):
     X_sample = preprocess_text(text)
     prediction = model.predict(X_sample)[0]
+    confidence = model.predict_proba(X_sample)[0][1]
     return prediction, confidence