likunchang commited on
Commit
468111e
·
1 Parent(s): c6b9a17
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -23,8 +23,8 @@ def edit_image(image: Image.Image, prompt: str, show_thinking=False, cfg_text_sc
23
  timestep_shift=3.0, num_timesteps=50, cfg_renorm_min=1.0,
24
  cfg_renorm_type="text_channel", max_think_token_n=1024,
25
  do_sample=False, text_temperature=0.3, seed=0):
26
-
27
- yield None, None
28
 
29
  # Helper function to load example images
30
  def load_example_image(image_path):
@@ -127,7 +127,7 @@ with gr.Blocks() as demo:
127
  )
128
 
129
  with gr.Column(scale=1):
130
- edit_image_output = gr.Image(label="Result")
131
  edit_thinking_output = gr.Textbox(label="Thinking Process", visible=False)
132
 
133
  with gr.Row():
 
23
  timestep_shift=3.0, num_timesteps=50, cfg_renorm_min=1.0,
24
  cfg_renorm_type="text_channel", max_think_token_n=1024,
25
  do_sample=False, text_temperature=0.3, seed=0):
26
+
27
+ yield (image, image), None
28
 
29
  # Helper function to load example images
30
  def load_example_image(image_path):
 
127
  )
128
 
129
  with gr.Column(scale=1):
130
+ edit_image_output = gr.ImageSlider(label="Result")
131
  edit_thinking_output = gr.Textbox(label="Thinking Process", visible=False)
132
 
133
  with gr.Row():