Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ from transformers import RobertaForSequenceClassification, RobertaTokenizer
|
|
9 |
sentiment_model = AutoModelForSequenceClassification.from_pretrained("SamanthaStorm/tether-sentiment")
|
10 |
sentiment_tokenizer = AutoTokenizer.from_pretrained("SamanthaStorm/tether-sentiment")
|
11 |
|
12 |
-
Load abuse pattern model
|
13 |
|
14 |
model_name = "SamanthaStorm/abuse-pattern-detector-v2"
|
15 |
model = RobertaForSequenceClassification.from_pretrained(model_name, trust_remote_code=True)
|
|
|
9 |
sentiment_model = AutoModelForSequenceClassification.from_pretrained("SamanthaStorm/tether-sentiment")
|
10 |
sentiment_tokenizer = AutoTokenizer.from_pretrained("SamanthaStorm/tether-sentiment")
|
11 |
|
12 |
+
# Load abuse pattern model
|
13 |
|
14 |
model_name = "SamanthaStorm/abuse-pattern-detector-v2"
|
15 |
model = RobertaForSequenceClassification.from_pretrained(model_name, trust_remote_code=True)
|