Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import torch
 import numpy as np
-from transformers import pipeline, RobertaForSequenceClassification, RobertaTokenizer
+from transformers import pipeline, AutoModelForSequenceClassification, AutoTokenizer
 from motif_tagging import detect_motifs
 import re
 import matplotlib.pyplot as plt
@@ -44,8 +44,8 @@ sst_pipeline = pipeline("sentiment-analysis", model="distilbert-base-uncased-fin
 
 # --- Abuse Model ---
 model_name = "SamanthaStorm/tether-multilabel-v1"
-model = RobertaForSequenceClassification.from_pretrained(model_name)
-tokenizer = RobertaTokenizer.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 LABELS = [
     "blame shifting", "contradictory statements", "control", "dismissiveness",