blanchon commited on
Commit
4ba226e
·
1 Parent(s): 457c1f0
Files changed (1) hide show
  1. app.py +22 -1
app.py CHANGED
@@ -525,7 +525,7 @@ with gr.Blocks(css=css) as demo:
525
  columns=2,
526
  height=500,
527
  format="png",
528
- # interactive=False,
529
  )
530
  run_button = gr.Button("Run")
531
  flag_button = gr.Button("Flag")
@@ -616,6 +616,27 @@ with gr.Blocks(css=css) as demo:
616
  "examples_dataset",
617
  )
618
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
619
 
620
  # demo.launch(auth=[(USER, PASSWORD)])
621
  demo.launch()
 
525
  columns=2,
526
  height=500,
527
  format="png",
528
+ interactive=False,
529
  )
530
  run_button = gr.Button("Run")
531
  flag_button = gr.Button("Flag")
 
616
  "examples_dataset",
617
  )
618
 
619
+ run_button.click(
620
+ fn=infer,
621
+ inputs=[
622
+ furniture_image_input,
623
+ room_image_input,
624
+ furniture_prompt,
625
+ seed,
626
+ randomize_seed,
627
+ guidance_scale,
628
+ num_inference_steps,
629
+ max_dimension,
630
+ num_images_per_prompt,
631
+ ],
632
+ outputs=[results, seed],
633
+ )
634
+ flag_button.click(
635
+ fn=flag,
636
+ inputs=[furniture_image_input, room_image_input, results],
637
+ preprocess=False,
638
+ )
639
+
640
 
641
  # demo.launch(auth=[(USER, PASSWORD)])
642
  demo.launch()