Update app.py

app.py

@@ -26,11 +26,13 @@ model = tf.keras.models.load_model('model')
 
 with st.form("form", clear_on_submit=True):
     uploaded_file = st.file_uploader("Upload image files", type="jpg", accept_multiple_files=True)
-    submitted = st.form_submit_button("Generate prodiction!")
-    
+    submitted = st.form_submit_button("Click to generate prediction")
+    if uploaded_file is not None:
+        st.image(uploaded_file, width=100)
+        
     if submitted and uploaded_file is not None:
         st.write("Predicting!")
-        st.image(uploaded_file, width=100)
+        
 
         for file in uploaded_file:
             img = Image.open(file)
@@ -42,4 +44,6 @@ with st.form("form", clear_on_submit=True):
             score = predictions[0]
             st.markdown("Predicted class of the image {} is : {}".format(file, class_names[np.argmax(score)]))
 
+        st.write("Click to clear all")
+        uploaded_file = None