Update app.py

app.py

@@ -173,6 +173,15 @@ def get_concat(ims):
     return dst
 
 
+def click_on_display(state):
+    if 'boxes' not in state:
+        state['boxes'] = []
+    boxes = state['boxes']
+    x = Image.open('./images/dog.png')
+    gen_images = [gr.Image.update(value=x, visible=True)]
+
+    return gen_images, state
+
 def generate(unet, vae, tokenizer, text_encoder, language_instruction, grounding_texts, sketch_pad,
              loss_threshold, guidance_scale, batch_size, rand_seed, max_step, loss_scale, max_iter,
              state):
@@ -507,7 +516,7 @@ def main():
                 queue=False)
 
             gen_btn.click(
-                fn=partial(generate, unet, vae, tokenizer, text_encoder),
+                fn=partial(click_on_display),
                 inputs=[
                     language_instruction, grounding_instruction, sketch_pad,
                     loss_threshold, guidance_scale, batch_size, rand_seed,