Update app.py

app.py

@@ -13,11 +13,11 @@ st.title("Photo Editor V1.0")
 
 def generate_image(image_path):
     image = load_image(image_path)
-    st.image(image, use_column_width=True, caption="Input Image")
+    st.image(image, use_column_width=True)
 
     st.write("Generated image:")
     generated_image = inpaint_image(image_path)
-    st.image(generated_image, use_column_width=True, caption="Output Image")
+    st.image(generated_image, use_column_width=True)
 
     # st.write("Generated image:")
     # generated_image = inpaint_image(image)
@@ -131,5 +131,4 @@ def inpaint_image(image_path):
 image_file = st.file_uploader("Upload an image", type=['jpg', 'png', 'jpeg'])
 
 if image_file is not None:
-    if st.button("Generate"):
-        generate_image(image_file.getvalue())
+    generate_image(image_file)