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)
+    st.image(image, use_column_width=True, caption="Original Image")
 
     st.write("Generated image:")
     generated_image = inpaint_image(image_path)
-    st.image(generated_image, use_column_width=True)
+    st.image(generated_image, use_column_width=True, caption="Generated Image")
 
     # st.write("Generated image:")
     # generated_image = inpaint_image(image)
@@ -104,7 +104,7 @@ def inpaint_image(image_path):
     url = "https://api.andersonrobotics.com/api/v1/generate"
 
     payload = {
-        "prompt": "a boy and a girl are playing with a ball and a cat and a dog and a cat",
+        "prompt": "a boy and a girl are playing with a ball and a cat and a dog",
         "num_samples": 1,
         "num_return": 1,
         "temperature": 1.0,
@@ -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)