Update app.py

app.py

@@ -102,5 +102,10 @@ if uploaded_file is not None:
             img_input = imread(input_path)
             img_output = imread(output_path)
             display(img_input, img_output)
+            #delete the uploaded file, input and output images
+            os.remove(file_path)
+            os.remove(input_path)
+            os.remove(output_path)
+            
     except Exception as e:
         st.error(f"An error occurred: {str(e)}")