Update app.py

app.py

@@ -31,7 +31,7 @@ uploaded_files = st.file_uploader(
     accept_multiple_files=True, 
     key=state['file_uploader_key'])
 
-if state['uploaded_files'] == []:
+if uploaded_files:
     st.image(uploaded_files, width=100)
     state["uploaded_files"] = uploaded_files
     if st.button("Generate prediction"):
@@ -44,10 +44,9 @@ if state['uploaded_files'] == []:
             predictions = model.predict(processed_image)
             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
-            state["uploaded_files"] = []
+            uploaded_files = None
             #st.experimental_rerun()