zs38 commited on
Commit
6f0d568
·
1 Parent(s): 1010ff1

fix example error

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -57,7 +57,7 @@ IMAGE_PROCESS_TRANSFORM = transforms.Compose([
57
  ])
58
 
59
  @spaces.GPU
60
- def generate_image(ref_image, ref_image2=None, prompt="", height=512, width=512, num_steps=25, guidance_scale=3.5, seed=0, ip_scale=1.0):
61
  print(f"ref_image: {ref_image.size if ref_image is not None else None}, "
62
  f"ref_image2: {ref_image2.size if ref_image2 is not None else None}, "
63
  f"prompt: {prompt}, height: {height}, width: {width}, num_steps: {num_steps}, guidance_scale: {guidance_scale}, ip_scale: {ip_scale}")
@@ -148,7 +148,7 @@ with gr.Blocks() as demo:
148
  # When the button is clicked, pass all inputs to generate_image
149
  generate_btn.click(
150
  fn=generate_image,
151
- inputs=[ref_image, ref_image2, description, height_slider, width_slider, steps_slider, guidance_slider, ref_scale_slider],
152
  outputs=output,
153
  )
154
 
 
57
  ])
58
 
59
  @spaces.GPU
60
+ def generate_image(ref_image, prompt="", height=512, width=512, ref_image2=None, num_steps=25, guidance_scale=3.5, seed=0, ip_scale=1.0):
61
  print(f"ref_image: {ref_image.size if ref_image is not None else None}, "
62
  f"ref_image2: {ref_image2.size if ref_image2 is not None else None}, "
63
  f"prompt: {prompt}, height: {height}, width: {width}, num_steps: {num_steps}, guidance_scale: {guidance_scale}, ip_scale: {ip_scale}")
 
148
  # When the button is clicked, pass all inputs to generate_image
149
  generate_btn.click(
150
  fn=generate_image,
151
+ inputs=[ref_image, description, height_slider, width_slider, ref_image2, steps_slider, guidance_slider, ref_scale_slider],
152
  outputs=output,
153
  )
154