Spaces:

valurank
/

spam_comment_detection

Runtime error

abdulmatinomotoso commited on Jul 30, 2024

Commit

846bf7a

verified ·

1 Parent(s): cdd14f3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ model_name = "valurank/distilroberta-spam-comments-detection"
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 def clean_text(raw_text):
   text = raw_text.encode("ascii", errors="ignore").decode(

 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 def clean_text(raw_text):
   text = raw_text.encode("ascii", errors="ignore").decode(