Benjamin S Liang commited on
Commit
8b71a2e
1 Parent(s): 1e67cf3

fixed app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -19,12 +19,9 @@ def load_model(opt):
19
  elif opt == 'twitter-roberta-base-sentiment':
20
  tokenizer = AutoTokenizer.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
21
  model = AutoModelForSequenceClassification.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
22
- elif opt == 'distilRoberta-financial-sentiment'
23
  tokenizer = AutoTokenizer.from_pretrained("mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis")
24
  model = AutoModelForSequenceClassification.from_pretrained("mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis")
25
- elif opt == 'bert-base-multilingual-uncased-sentiment ':
26
- tokenizer = AutoTokenizer.from_pretrained("nlptown/bert-base-multilingual-uncased-sentiment")
27
- model = AutoModelForSequenceClassification.from_pretrained("nlptown/bert-base-multilingual-uncas
28
 
29
  elif not model and not tokenizer:
30
  print("Model not loaded correctly. Try again!")
 
19
  elif opt == 'twitter-roberta-base-sentiment':
20
  tokenizer = AutoTokenizer.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
21
  model = AutoModelForSequenceClassification.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
22
+ elif opt == 'distilRoberta-financial-sentiment':
23
  tokenizer = AutoTokenizer.from_pretrained("mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis")
24
  model = AutoModelForSequenceClassification.from_pretrained("mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis")
 
 
 
25
 
26
  elif not model and not tokenizer:
27
  print("Model not loaded correctly. Try again!")