Spaces:

inkleaves
/

spam_detection_app

Sleeping

inkleaves commited on Feb 26

Commit

6bd241c

verified ·

1 Parent(s): ea4a35e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,5 +10,5 @@ def predict(text):
     prediction = outputs.logits.argmax(dim=-1).item()
     return "Spam" if prediction == 1 else "Not Spam"
-interface = gr.Interface(fn=predict, inputs="Text", outputs="Text")
 interface.launch(share=True)

     prediction = outputs.logits.argmax(dim=-1).item()
     return "Spam" if prediction == 1 else "Not Spam"
+interface = gr.Interface(fn=predict, inputs="text", outputs="text")
 interface.launch(share=True)