added button

app.py

@@ -28,16 +28,31 @@ def model_prediction(image, model):
     else:
         result = "Prediction is control"
     return result
-        
-if file is None:
-    st.text("Please upload an image file")
-else:
-    image = Image.open(file)
-    st.image(image, use_column_width=True)
-    new_image = np.stack((image,), axis=0)
-    predictions = model_prediction(new_image, model)
-    st.write(prediction)
-    print(
-    "This image most likely belongs to {}."
-    .format(prediction)
+
+def model_prediction(image, model):
+    img = cv2.imread(image)
+    plt.imshow(img)
+    plt.show()
+    resize = tf.image.resize(img, (256,256))
+    yhat = model.predict(np.expand_dims(resize/255, 0))
+    if(yhat>0.5):
+        result = "Prediction is loose"
+    else:
+        result = "Prediction is control"
+    return result
+
+
+def on_click():
+    if file is None:
+        st.text("Please upload an image file")
+    else:
+        image = Image.open(file)
+        st.image(image, use_column_width=True)
+        predictions = model_prediction(image, model)
+        st.write(prediction)
+        print(
+            "This image most likely belongs to {}."
+            .format(prediction)
 )
+
+st.button('Predict', on_click=on_click)