gradio slide

app.py

@@ -1,6 +1,8 @@
 
 import gradio as gr
 from PIL import Image
+from gradio_imageslider import ImageSlider
+
 import requests
 import base64
 import numpy as np
@@ -22,6 +24,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 
 
 def generate(
+    input_image: Image,
     prompt: str,
     negative_prompt: str = "",
     seed: int = 0,
@@ -39,9 +42,9 @@ def generate(
     payload = {
         "input": {
             "hdr": 0,
-            "image": "https://replicate.delivery/pbxt/KA9yP9n3ZX5A5mkoPz3gsPzKTH1NA7LqVkQRTg7Sov46lOfo/0_1.webp",
+            "image": "http://localhost:7860/file=" + input_image,
             "steps": 20,
-            "prompt": "UHD 4k vogue, a woman resting in a magic pool, face above the surface of the water, red freckles",
+            "prompt": prompt,
             "scheduler": "DDIM",
             "creativity": 0.25,
             "guess_mode": False,
@@ -58,29 +61,34 @@ def generate(
         image_data = base64.b64decode(
             base64_image.replace("data:image/png;base64,", ""))
         image_stream = io.BytesIO(image_data)
-        return Image.open(image_stream)
+        return [Image.open(input_image), Image.open(image_stream)]
     raise gr.Error(json_response["status"])
 
 
 examples = [
-    "An astronaut riding a green horse",
-    "A mecha robot in a favela by Tarsila do Amaral",
-    "The sprirt of a Tamagotchi wandering in the city of Los Angeles",
-    "A delicious feijoada ramen dish"
+    ["An astronaut riding a green horse", "examples/image2.png"],
+    ["A mecha robot in a favela by Tarsila do Amaral", "examples/image2.png"],
+    ["The sprirt of a Tamagotchi wandering in the city of Los Angeles",
+        "examples/image1.png"],
+    ["A delicious feijoada ramen dish", "examples/image0.png"],
 ]
 
 with gr.Blocks() as demo:
-    with gr.Group():
-        with gr.Row():
-            prompt = gr.Text(
-                label="Prompt",
-                show_label=False,
-                max_lines=1,
-                placeholder="Enter your prompt",
-                container=False,
-            )
-            run_button = gr.Button("Run", scale=0)
-        result = gr.Image(label="Result", show_label=False)
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(type="filepath")
+            with gr.Group():
+                with gr.Row():
+                    prompt = gr.Text(
+                        label="Prompt",
+                        show_label=False,
+                        max_lines=1,
+                        placeholder="Enter your prompt",
+                        container=False,
+                    )
+                    run_button = gr.Button("Run", scale=0)
+        with gr.Column():
+            result = ImageSlider(label="Result", type="pil")
     with gr.Accordion("Advanced options", open=False):
         negative_prompt = gr.Text(
             label="Negative prompt",
@@ -151,13 +159,14 @@ with gr.Blocks() as demo:
 
     gr.Examples(
         examples=examples,
-        inputs=prompt,
+        inputs=[prompt, input_image],
         outputs=result,
         fn=generate,
-        cache_examples=False,
+        cache_examples=True,
     )
 
     inputs = [
+        input_image,
         prompt,
         negative_prompt,
         seed,

requirements.txt

@@ -1,2 +1,3 @@
 gradio==4.18.0
-replicate==0.23.1
+replicate==0.23.1
+gradio_imageslider==0.0.17