Spaces:

resolverkatla
/

Spam_Detector

Runtime error

resolverkatla commited on 20 days ago

Commit

d2faca8

1 Parent(s): ffe55c8

.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 # Load the spam classifier model
-classifier = pipeline("text-classification", model="dima806/email-spam-detection-roberta")
 def spam_detector(text):
     result = classifier(text)

 from transformers import pipeline
 # Load the spam classifier model
+classifier = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-sms-spam-detection")
 def spam_detector(text):
     result = classifier(text)