Spaces:

shrushhtijadhav
/

spam-detector

Sleeping

shrushhtijadhav commited on Jun 11

Commit

19d7e2f

verified ·

1 Parent(s): 53cb605

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 from transformers import pipeline
 import gradio as gr
-classifier = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-sms-spam-detection")
 label_map = {
     "LABEL_0": "HAM (Not Spam)",

 from transformers import pipeline
 import gradio as gr
+classifier = pipeline("text-classification", model="bhadresh-savani/bert-base-uncased-emotion")
 label_map = {
     "LABEL_0": "HAM (Not Spam)",