Spaces:

vives
/

bert_cvent_top_k_sim

Runtime error

vives commited on Jun 1, 2022

Commit

58bd5f2

1 Parent(s): 8ff8764

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,8 +14,8 @@ kp_cosine_checkpoint = "cosine_kp.pickle"
 model_finbert = AutoModelForSequenceClassification.from_pretrained("ProsusAI/finbert", output_hidden_states=True)
 tokenizer_finbert = AutoTokenizer.from_pretrained("ProsusAI/finbert")
-kp_dict_finbert_checkpoint = "kp_dict_finbert.pickle"
-kp_cosine_finbert_checkpoint = "cosine_kp_finbert.pickle"
 tokenizer_sapbert = AutoTokenizer.from_pretrained("cambridgeltl/SapBERT-from-PubMedBERT-fulltext")
 model_sapbert = AutoModel.from_pretrained("cambridgeltl/SapBERT-from-PubMedBERT-fulltext", output_hidden_states=True)

 model_finbert = AutoModelForSequenceClassification.from_pretrained("ProsusAI/finbert", output_hidden_states=True)
 tokenizer_finbert = AutoTokenizer.from_pretrained("ProsusAI/finbert")
+kp_dict_finbert_checkpoint = "kp_dict_finance.pickle"
+kp_cosine_finbert_checkpoint = "cosine_kp_finance.pickle"
 tokenizer_sapbert = AutoTokenizer.from_pretrained("cambridgeltl/SapBERT-from-PubMedBERT-fulltext")
 model_sapbert = AutoModel.from_pretrained("cambridgeltl/SapBERT-from-PubMedBERT-fulltext", output_hidden_states=True)