Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
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)
|