SamanthaStorm commited on
Commit
6cdcb5b
·
verified ·
1 Parent(s): e46fbeb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -43,9 +43,11 @@ def generate_abuse_score_chart(dates, scores, labels):
43
  sst_pipeline = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
44
 
45
  # --- Abuse Model ---
46
- model_name = "SamanthaStorm/autotrain-jlpi4-mllvp"
47
- model = RobertaForSequenceClassification.from_pretrained(model_name)
48
- tokenizer = RobertaTokenizer.from_pretrained(model_name)
 
 
49
 
50
  LABELS = [
51
  "blame shifting", "contradictory statements", "control", "dismissiveness",
 
43
  sst_pipeline = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
44
 
45
  # --- Abuse Model ---
46
+ from transformers import AutoModelForSequenceClassification, AutoTokenizer
47
+
48
+ model_name = "SamanthaStorm/tether-multilabel-v2"
49
+ model = AutoModelForSequenceClassification.from_pretrained(model_name)
50
+ tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
51
 
52
  LABELS = [
53
  "blame shifting", "contradictory statements", "control", "dismissiveness",