Update app.py

app.py

@@ -152,6 +152,7 @@ if uploaded_file is not None:
             st.markdown("""<style>[data-baseweb="select"] {margin-top: -36px;}</style>""", unsafe_allow_html=True)
             threshold = st.slider("", 0.0, 1.0, 0.0, 0.05, label_visibility="hidden")
             y_pred = np.where(y_pred > threshold, y_pred, 0)
+            plot_prediction(y_pred)
 
         plot_prediction(y_pred)