SamanthaStorm commited on
Commit
52f19d2
·
verified ·
1 Parent(s): 394b179

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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
- tokenizer = AutoTokenizer.from_pretrained("SamanthaStorm/tether-sentiment")
73
- model = AutoModelForSequenceClassification.from_pretrained("SamanthaStorm/tether-sentiment")
 
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",