Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
47 |
-
|
48 |
-
|
|
|
|
|
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",
|