Update app.py

app.py

@@ -12,14 +12,16 @@ if choice == "distilbert-cvent":
   model_checkpoint = "vives/distilbert-base-uncased-finetuned-cvent-2019_2022"
   model = AutoModelForMaskedLM.from_pretrained(model_checkpoint, output_hidden_states=True)
   tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
-  kp_dict_checkpoint = "kp_dict_finbert.pickle"
-  kp_cosine_checkpoint = "cosine_kp_finbert.pickle"
+  kp_dict_checkpoint = "kp_dict_merged.pickle"
+  kp_cosine_checkpoint = "cosine_kp.pickle"
+   
 elif choice == "finbert":
   model_checkpoint = "ProsusAI/finbert"
   tokenizer = AutoTokenizer.from_pretrained("ProsusAI/finbert")
   model = AutoModelForSequenceClassification.from_pretrained("ProsusAI/finbert", output_hidden_states=True)
-  kp_dict_checkpoint = "kp_dict_merged.pickle"
-  kp_cosine_checkpoint = "cosine_kp.pickle"
+  kp_dict_checkpoint = "kp_dict_finbert.pickle"
+  kp_cosine_checkpoint = "cosine_kp_finbert.pickle"
+
 text = st.text_input("Enter word or key-phrase")
 
 exclude_words = st.radio("exclude_words",[True,False], help="Exclude results that contain any words in the query (i.e exclude 'hot coffee' if the query is 'cold coffee')")