SamanthaStorm commited on
Commit
92b1518
·
verified ·
1 Parent(s): 6cbeaf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,8 +18,8 @@ EMOTION_TO_SENTIMENT = {
18
 
19
  # --- Abuse Model ---
20
  model_name = "SamanthaStorm/autotrain-jlpi4-mllvp"
21
- model = RobertaForSequenceClassification.from_pretrained(model_name, trust_remote_code=True)
22
- tokenizer = RobertaTokenizer.from_pretrained(model_name, trust_remote_code=True)
23
 
24
  LABELS = [
25
  "blame shifting", "contradictory statements", "control", "dismissiveness",
 
18
 
19
  # --- Abuse Model ---
20
  model_name = "SamanthaStorm/autotrain-jlpi4-mllvp"
21
+ model = RobertaForSequenceClassification.from_pretrained(model_name)
22
+ tokenizer = RobertaTokenizer.from_pretrained(model_name)
23
 
24
  LABELS = [
25
  "blame shifting", "contradictory statements", "control", "dismissiveness",