Upload app.py

app.py

@@ -80,7 +80,6 @@ if __name__ == "__main__":
                                 st.code(c)
                         st.markdown("### Vulnerable Function:")
                 st.code(input_code, language="cpp", line_numbers=True)
-                st.snow()
                     
     elif behavior == "Analyze my own":
         # user input of project title
@@ -98,8 +97,6 @@ if __name__ == "__main__":
                     line_pred = out["batch_statement_pred"][0]
                     line_confidence = out["batch_statement_pred_prob"][0]      
                 output = None
-                # inference complete
-                st.snow()
                 print_code = input_code.split("\n")[:MAX_NUM_STATEMENTS]
                 if func_pred == 0:
                     st.markdown("### Scanning Results:")
@@ -116,4 +113,3 @@ if __name__ == "__main__":
                                 st.code(c)
                         st.markdown("### Vulnerable Function:")   
                 st.code(input_code, language="cpp", line_numbers=True)
-                st.snow()