Update app.py

app.py

@@ -182,7 +182,7 @@ def click_on_display(language_instruction, grounding_texts, sketch_pad,
     x = Image.open('./images/dog.png')
     gen_images = [gr.Image.update(value=x, visible=True)]
 
-    return gen_images, state
+    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,