Spaces:
Runtime error
Runtime error
Benjamin S Liang
commited on
Commit
•
8b71a2e
1
Parent(s):
1e67cf3
fixed app.py
Browse files
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!")
|