vives commited on
Commit
06e298f
1 Parent(s): 3abc396

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ if choice == "distilbert-cvent":
12
  model_checkpoint = "vives/distilbert-base-uncased-finetuned-cvent-2019_2022"
13
  model = AutoModelForMaskedLM.from_pretrained(model_checkpoint, output_hidden_states=True)
14
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
15
- else if choice == "finbert":
16
  model_checkpoint = "ProsusAI/finbert"
17
  tokenizer = AutoTokenizer.from_pretrained("ProsusAI/finbert")
18
  model = AutoModelForSequenceClassification.from_pretrained("ProsusAI/finbert", output_hidden_states=True)
 
12
  model_checkpoint = "vives/distilbert-base-uncased-finetuned-cvent-2019_2022"
13
  model = AutoModelForMaskedLM.from_pretrained(model_checkpoint, output_hidden_states=True)
14
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
15
+ elif choice == "finbert":
16
  model_checkpoint = "ProsusAI/finbert"
17
  tokenizer = AutoTokenizer.from_pretrained("ProsusAI/finbert")
18
  model = AutoModelForSequenceClassification.from_pretrained("ProsusAI/finbert", output_hidden_states=True)