Update app.py

app.py

@@ -26,9 +26,10 @@ 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("Click to generate prediction")
     if uploaded_file is not None:
         st.image(uploaded_file, width=100)
+    
+    submitted = st.form_submit_button("Click to generate prediction")
         
     if submitted and uploaded_file is not None:
         st.write("Predicting!")
@@ -44,6 +45,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
+