Update app.py

app.py

@@ -15,9 +15,9 @@ def generate_image(image_path):
     image = load_image(image_path)
     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)
+    st.write("Generated image:")
+    generated_image = inpaint_image(image_path)
+    st.image(generated_image, use_column_width=True)
 
     # st.write("Generated image:")
     # generated_image = inpaint_image(image)
@@ -131,4 +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:
-    generate_image(image_file)
+    generate_image(image_file)