Fix

app.py

@@ -136,8 +136,8 @@ def run_test(model_names,model_name,sentences,display_area,main_index,user_uploa
 
 def display_results(orig_sentences,main_index,results,response_info,app_mode):
     main_sent = f"<div style=\"font-size:14px; color: #2f2f2f; text-align: left\">{response_info}<br/><br/></div>"
-    score_text = "cosine_distance" if app_mode == "similarity" else "cosine_distance/score"
-    pivot_name = "main sentence" if app_mode == "similarity" else "query"
+    score_text = "cosine_distance" if app_mode == SEM_SIMILARITY else "cosine_distance/score"
+    pivot_name = "main sentence" if app_mode == SEM_SIMILARITY else "query"
     main_sent += f"<div style=\"font-size:14px; color: #6f6f6f; text-align: left\">Results sorted by {score_text}. Closest to furthest away from {pivot_name}</div>"
     pivot_name = pivot_name[0].upper() + pivot_name[1:]
     main_sent += f"<div style=\"font-size:16px; color: #2f2f2f; text-align: left\"><b>{pivot_name}:</b>&nbsp;&nbsp;{orig_sentences[main_index]}</div>"
@@ -192,7 +192,7 @@ def app_main(app_mode,example_files,model_name_files):
         selected_model = st.selectbox(label=selection_label,  
                     options = options_arr, index=0,  key = "twc_model")
         st.write("")
-        if (app_mode == "similarity"):
+        if (app_mode == SEM_SIMILARITY):
             main_index = st.number_input('Step 3. Enter index of sentence in file to make it the main sentence',value=1,min_value = 1)
         else:
             main_index = 1