Update app.py

app.py

@@ -160,8 +160,16 @@ with gr.Blocks(css=css) as demo:
                 return f"Failed to load image from URL: {e}"
         return None
 
-    use_generated_image.click(fn=lambda: select_image(None, None, True), inputs=[], outputs=additional_image_output)
-    uploaded_image.change(fn=select_image, inputs=[uploaded_image, image_url, gr.State(False)], outputs=additional_image_output)
-    image_url.submit(fn=select_image, inputs=[uploaded_image, image_url, gr.State(False)], outputs=additional_image_output)
+    @spaces.GPU
+    def image2image(uploaded_image, image_url, use_generated=False):
+        image = select_image(uploaded_image, image_url, use_generated=False)
+        prompt = "ghibli style, a fantasy landscape with castles"
+        generator = torch.Generator(device=device).manual_seed(1024)
+        image = pipe(prompt=prompt, image=image, strength=0.75, guidance_scale=7.5, generator=generator).images[0]
+        return image
+    
+    use_generated_image.click(fn=lambda: image2image(None, None, True), inputs=[], outputs=additional_image_output)
+    uploaded_image.change(fn=image2image, inputs=[uploaded_image, image_url, gr.State(False)], outputs=additional_image_output)
+    image_url.submit(fn=image2image, inputs=[uploaded_image, image_url, gr.State(False)], outputs=additional_image_output)
 
 demo.launch()