Update appStore/target.py

appStore/target.py

@@ -81,7 +81,7 @@ def app():
             st.session_state.key2 = df
 
 
-def target_display(): 
+def target_display(model_sel_name): 
     
     ### TABLE Output ###
 
@@ -107,8 +107,7 @@ def target_display():
     # construct RAG query for each label, send to openai and process response
     for i in range(0,len(df_agg)):
         st.write(df_agg['Vulnerability Label'].iloc[i])
-        run_query(context = df_agg['text'].iloc[i], label = df_agg['Vulnerability Label'].iloc[i])
-        # st.write(df_agg['text'].iloc[i])
+        run_query(context = df_agg['text'].iloc[i], label = df_agg['Vulnerability Label'].iloc[i], model_sel_name=model_sel_name)