Benjamin S Liang commited on
Commit
96db83a
1 Parent(s): 03251df
Files changed (1) hide show
  1. app.py +1 -1
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
- else if opt == DEFAULT_OPT:
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")