SamanthaStorm commited on
Commit
ba75107
·
verified ·
1 Parent(s): 96f05f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
29
  sentiment_model = AutoModelForSequenceClassification.from_pretrained("SamanthaStorm/tether-sentiment").to(device)
30
  sentiment_tokenizer = AutoTokenizer.from_pretrained("SamanthaStorm/tether-sentiment", use_fast=False)
31
 
32
- emotion_pipeline = pipeline(
33
  "text-classification",
34
  model="j-hartmann/emotion-english-distilroberta-base",
35
  tokenizer="j-hartmann/emotion-english-distilroberta-base",
 
29
  sentiment_model = AutoModelForSequenceClassification.from_pretrained("SamanthaStorm/tether-sentiment").to(device)
30
  sentiment_tokenizer = AutoTokenizer.from_pretrained("SamanthaStorm/tether-sentiment", use_fast=False)
31
 
32
+ emotion_pipeline = hf_pipeline(
33
  "text-classification",
34
  model="j-hartmann/emotion-english-distilroberta-base",
35
  tokenizer="j-hartmann/emotion-english-distilroberta-base",