SamanthaStorm commited on
Commit
950da18
·
verified ·
1 Parent(s): e01a07a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -74,12 +74,12 @@ tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
74
 
75
  # sentiment model
76
  sentiment_model = AutoModelForSequenceClassification.from_pretrained(
77
- "SamanthaStorm/tether-sentiment",
78
  force_download=True,
79
  local_files_only=False
80
  ).to(device)
81
  sentiment_tokenizer = AutoTokenizer.from_pretrained(
82
- "SamanthaStorm/tether-sentiment",
83
  use_fast=False,
84
  force_download=True,
85
  local_files_only=False
@@ -112,7 +112,7 @@ LABELS = [
112
  "contradictory statements", "obscure language"
113
  ]
114
 
115
- SENTIMENT_LABELS = ["undermining", "supportive"]
116
 
117
  THRESHOLDS = {
118
  "recovery phase": 0.324,
 
74
 
75
  # sentiment model
76
  sentiment_model = AutoModelForSequenceClassification.from_pretrained(
77
+ "SamanthaStorm/tether-sentiment-v3",
78
  force_download=True,
79
  local_files_only=False
80
  ).to(device)
81
  sentiment_tokenizer = AutoTokenizer.from_pretrained(
82
+ "SamanthaStorm/tether-sentiment-v3",
83
  use_fast=False,
84
  force_download=True,
85
  local_files_only=False
 
112
  "contradictory statements", "obscure language"
113
  ]
114
 
115
+ SENTIMENT_LABELS = ["supportive", "undermining"]
116
 
117
  THRESHOLDS = {
118
  "recovery phase": 0.324,