Spaces:
Runtime error
Runtime error
benliang99
commited on
Commit
•
96db83a
1
Parent(s):
03251df
Fix elif
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ def load_model(opt):
|
|
15 |
if opt == FINETUNED_OPT:
|
16 |
tokenizer = DistilBertTokenizerFast.from_pretrained('distilbert-base-uncased')
|
17 |
model = DistilBertForSequenceClassification.from_pretrained('saccharinedreams/finetuned-distilbert-base-uncased-for-hupd')
|
18 |
-
|
19 |
return model, tokenizer
|
20 |
elif opt == 'bertweet-base-sentiment-analysis':
|
21 |
tokenizer = AutoTokenizer.from_pretrained("finiteautomata/bertweet-base-sentiment-analysis")
|
|
|
15 |
if opt == FINETUNED_OPT:
|
16 |
tokenizer = DistilBertTokenizerFast.from_pretrained('distilbert-base-uncased')
|
17 |
model = DistilBertForSequenceClassification.from_pretrained('saccharinedreams/finetuned-distilbert-base-uncased-for-hupd')
|
18 |
+
elif opt == DEFAULT_OPT:
|
19 |
return model, tokenizer
|
20 |
elif opt == 'bertweet-base-sentiment-analysis':
|
21 |
tokenizer = AutoTokenizer.from_pretrained("finiteautomata/bertweet-base-sentiment-analysis")
|