Update app.py

app.py

@@ -31,7 +31,7 @@ uploaded_files = st.file_uploader(
     accept_multiple_files=True, 
     key=state['file_uploader_key'])
 
-if uploaded_files:
+if state['uploaded_files'] == []:
     st.image(uploaded_files, width=100)
     state["uploaded_files"] = uploaded_files
     if st.button("Generate prediction"):
@@ -46,7 +46,8 @@ if uploaded_files:
             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
-    st.experimental_rerun()
+        if st.button("Clear uploaded files"):
+            state["file_uploader_key"] += 1
+            state["uploaded_files"] = []
+            #st.experimental_rerun()