Update app.py

app.py

@@ -155,12 +155,10 @@ if uploaded_file is not None:
 
         np.save("pred.npy", y_pred)
 
-    if bool(threshold):
-        y_pred = np.load("pred.npy")
+    y_pred = np.load("pred.npy")    
+    y_pred = np.where(y_pred > threshold, y_pred, 0)
         
-        pred = np.where(y_pred > threshold, y_pred, 0)
-        
-    plot_prediction(pred)
+    plot_prediction(y_pred)
 
     # with colC:
     #     st.markdown("""<style>[data-baseweb="select"] {margin-top: -36px;}</style>""", unsafe_allow_html=True)