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