prithivMLmods commited on
Commit
5023417
·
verified ·
1 Parent(s): 4413ed7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import gradio as gr
2
  import spaces
3
  import torch
4
- from gradio_imageslider import ImageSlider
5
  from diffusers import AutoencoderKL, TCDScheduler
6
  from diffusers.models.model_loading_utils import load_state_dict
7
  from huggingface_hub import hf_hub_download
@@ -348,7 +347,7 @@ with gr.Blocks(css=css) as demo:
348
  )
349
 
350
  with gr.Column():
351
- result = ImageSlider(label="Generated Image", interactive=False)
352
  use_as_input_button = gr.Button("Use as Input Image", visible=False)
353
 
354
  history_gallery = gr.Gallery(label="History", columns=6, object_fit="contain", interactive=False)
 
1
  import gradio as gr
2
  import spaces
3
  import torch
 
4
  from diffusers import AutoencoderKL, TCDScheduler
5
  from diffusers.models.model_loading_utils import load_state_dict
6
  from huggingface_hub import hf_hub_download
 
347
  )
348
 
349
  with gr.Column():
350
+ result = gr.Image(label="Generated Image", type="pil")
351
  use_as_input_button = gr.Button("Use as Input Image", visible=False)
352
 
353
  history_gallery = gr.Gallery(label="History", columns=6, object_fit="contain", interactive=False)