Update app.py

app.py

@@ -181,7 +181,7 @@ if 'slider' not in st.session_state:
     st.session_state['slider'] = 0
 
 if 'radio' not in st.session_state:
-    st.session_state['radio'] = 'CivileLaw-IR'
+    st.session_state['radio'] = 'Civile-Law-IR'
 
 if 'show' not in st.session_state:
     st.session_state['show'] = False
@@ -217,18 +217,18 @@ def run_inference(model_name, query):
 
     # Compute the similarity scores for these combinations
 
-    if model_name=='CivileLaw-IR':
+    if model_name=='Civile-Law-IR':
         similarity_scores = model_nli.predict(sentence_combinations)
         scores = [(score_max[0],idx) for idx,score_max in enumerate(similarity_scores)]
         sim_scores_argsort = sorted(scores, key=lambda x: x[0], reverse=True)
         results = [pred[idx] for _,idx in list(sim_scores_argsort)[:int(top_K)]]
 
-    if model_name=='STSB-IR':
+    if model_name=='STSB':
         similarity_scores = model_nli_stsb.predict(sentence_combinations)
         sim_scores_argsort = reversed(np.argsort(similarity_scores))
         results = [pred[idx] for idx in list(sim_scores_argsort)[:int(top_K)]]
     
-    if model_name=='NoRank-Baseline':
+    if model_name=='DR-Baseline':
         similarity_scores = model_baseline.predict(sentence_combinations)
         scores = [(score_max[0],idx) for idx,score_max in enumerate(similarity_scores)]
         sim_scores_argsort = sorted(scores, key=lambda x: x[0], reverse=True)
@@ -254,8 +254,8 @@ top_K = st.text_input('Choose Number of Result: ','10')
 
 model_name = st.radio(
         "Choose Model",
-        ("CivileLaw-IR", "STSB-IR", "NoRank-Baseline"),
-         key='radio', on_change=callback, args=('radio','CivileLaw-IR')
+        ("Civile-Law-IR", "STSB", "DR-Baseline"),
+         key='radio', on_change=callback, args=('radio','CivileLaw-IR'), help="Civile-Law-IR: trained on civile-NLI-dataset, STSB: trained on STSB french dataset, DR-Baseline: existing nli model trained on ms marco dataset"
     )
 
 
@@ -266,7 +266,7 @@ if st.button('Run', key='run'):
     st.session_state['show'] = True
     st.session_state['results'] = results
     st.session_state['query'] = query
-    model_dict = {'CivileLaw-IR': 'NLI-Syn', 'STSB-IR': 'NLI-stsb', 'NoRank-Baseline': 'NLI-baseline'}
+    model_dict = {'Civile-Law-IR': 'NLI-Syn', 'STSB': 'NLI-stsb', 'DR-Baseline': 'NLI-baseline'}
     st.session_state['model'] = model_dict[model_name]