Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -66,11 +66,10 @@ 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 |
-
# Load model directly
|
70 |
-
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
71 |
|
72 |
-
|
73 |
-
|
|
|
74 |
|
75 |
emotion_pipeline = hf_pipeline(
|
76 |
"text-classification",
|
|
|
66 |
model = AutoModelForSequenceClassification.from_pretrained(model_name).to(device)
|
67 |
tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
|
68 |
|
|
|
|
|
69 |
|
70 |
+
|
71 |
+
tokenizer = AutoTokenizer.from_pretrained("SamanthaStorm/tether-sentiment-v2")
|
72 |
+
model = AutoModelForSequenceClassification.from_pretrained("SamanthaStorm/tether-sentiment-v2")
|
73 |
|
74 |
emotion_pipeline = hf_pipeline(
|
75 |
"text-classification",
|