SamanthaStorm commited on
Commit
394b179
·
verified ·
1 Parent(s): 6c8218a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
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
- # 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
 
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",