Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -66,9 +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 |
-
|
71 |
-
|
|
|
|
|
72 |
|
73 |
emotion_pipeline = hf_pipeline(
|
74 |
"text-classification",
|
|
|
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 |
+
tokenizer = AutoTokenizer.from_pretrained("SamanthaStorm/tether-sentiment")
|
73 |
+
model = AutoModelForSequenceClassification.from_pretrained("SamanthaStorm/tether-sentiment")
|
74 |
|
75 |
emotion_pipeline = hf_pipeline(
|
76 |
"text-classification",
|