Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -66,10 +66,11 @@ model_name = "SamanthaStorm/tether-multilabel-v4"
|
|
| 66 |
model = AutoModelForSequenceClassification.from_pretrained(model_name).to(device)
|
| 67 |
tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
|
| 68 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 69 |
|
| 70 |
-
model_name = "SamanthaStorm/tether-sentiment"
|
| 71 |
-
tokenizer = AutoTokenizer.from_pretrained("SamanthaStorm/tether-sentiment")
|
| 72 |
-
model = AutoModelForSequenceClassification.from_pretrained("SamanthaStorm/tether-sentiment")
|
| 73 |
|
| 74 |
emotion_pipeline = hf_pipeline(
|
| 75 |
"text-classification",
|
|
|
|
| 66 |
model = AutoModelForSequenceClassification.from_pretrained(model_name).to(device)
|
| 67 |
tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
|
| 68 |
|
| 69 |
+
# sentiment model
|
| 70 |
+
sentiment_model = AutoModelForSequenceClassification.from_pretrained("SamanthaStorm/tether-sentiment").to(device)
|
| 71 |
+
sentiment_tokenizer = AutoTokenizer.from_pretrained("SamanthaStorm/tether-sentiment", use_fast=False)
|
| 72 |
+
sentiment_model.eval()
|
| 73 |
|
|
|
|
|
|
|
|
|
|
| 74 |
|
| 75 |
emotion_pipeline = hf_pipeline(
|
| 76 |
"text-classification",
|