Update app.py

app.py

@@ -34,6 +34,12 @@ uploaded_files = st.file_uploader(
 
 if uploaded_files:
     state["uploaded_files"] = uploaded_files
+
+    if st.button("Clear all"):
+        state["file_uploader_key"] += 1
+        time.sleep(.5)
+        st.experimental_rerun()
+
     if st.button("Generate prediction"):
         for file in uploaded_files:
             img = Image.open(file)
@@ -45,8 +51,4 @@ if uploaded_files:
             score = predictions[0]
             st.markdown("Predicted class of the image {} is : {}".format(file, class_names[np.argmax(score)]))
             
-    if st.button("Clear uploaded files"):
-        state["file_uploader_key"] += 1
-        time.sleep(.5)
-        st.experimental_rerun()