temp fix

kamwoh · Dec 19, 2023 · 63a83a4 · 63a83a4
1 parent 6a3e697
commit 63a83a4
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/gradio_demo_cub200.py b/src/gradio_demo_cub200.py
@@ -4,10 +4,11 @@
 import gradio as gr
 import torch
 from PIL import Image
-from gradio_imageslider import ImageSlider
 
 from dreamcreature.pipeline import create_args, load_pipeline
 
+# from gradio_imageslider import ImageSlider
+
 MAPPING = {
     'body': 0,
     'tail': 1,
@@ -79,10 +80,10 @@ def generate_images(prompt, negative_prompt, num_inference_steps, guidance_scale
                 seed = gr.Number(label="Seed", value=42)
                 button = gr.Button()
 
-            output_images = ImageSlider(show_label=False)
+            # output_images = ImageSlider(show_label=False)
 
     button.click(fn=generate_images,
                  inputs=[prompt, negative_prompt, num_inference_steps, guidance_scale,
-                         seed], outputs=[output_images], show_progress=True)
+                         seed], outputs='image', show_progress=True)
 
 demo.queue().launch(inline=False, share=True, debug=True)