Update appStore/vulnerability_analysis.py

appStore/vulnerability_analysis.py

@@ -73,6 +73,10 @@ def app():
                 df = vulnerability_classification(haystack_doc=df,
                                             threshold= params['threshold'])
 
+                # New SetFit script gives output as tensor, turn into right format
+                st.write("This is the output")
+                st.write(df)
+
 
                 # Store df in session state with key1
                 st.session_state.key1 = df