Update app.py

app.py

@@ -1,13 +1,15 @@
 import gradio as gr
 import torch
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from transformers import RobertaForSequenceClassification, RobertaTokenizer
 import numpy as np
 import tempfile
 
-# Load your updated model and tokenizer from Hugging Face
+# Load model and tokenizer
 model_name = "SamanthaStorm/abuse-pattern-detector-v2"
-model = AutoModelForSequenceClassification.from_pretrained(model_name, force_download=True)
-tokenizer = AutoTokenizer.from_pretrained(model_name, force_download=True)
+model = RobertaForSequenceClassification.from_pretrained(model_name)
+tokenizer = RobertaTokenizer.from_pretrained(model_name)
+
+TOTAL_LABELS = 17
 
 # Our model outputs 17 labels:
 # - First 14 are abuse pattern categories