Update app.py

app.py

@@ -320,7 +320,10 @@ def result_page():
         st.error("Image or mask file not found for overlay.")
 
     if st.button('Back to Upload'):
-        
+        for file in os.listdir(PATCHES_DIR):
+            os.remove(os.path.join(PATCHES_DIR, file))
+        for file in os.listdir(PRED_PATCHES_DIR):
+            os.remove(os.path.join(PRED_PATCHES_DIR, file))
         st.session_state.page = 'upload'
         st.session_state.file_uploaded = False
         st.session_state.filename = None