Update app.py

app.py

@@ -148,12 +148,16 @@ def respond(message, history):
                         yield output
             elif json_data["name"] == "image_generation":
                 query = json_data["arguments"]["query"]
+                number_of_image = json_data["arguments"]["number_of_image"]
                 gr.Info("Generating Image, Please wait 10 sec...")
                 client = InferenceClient("stabilityai/stable-diffusion-3-medium-diffusers")
-                seed = random.randint(0,999999)
-                image = client.text_to_image(message_text, negative_prompt=f"{seed}", num_inference_steps=20 )
-                yield gr.Image(image)
-                gr.Info("We are going to mor upgrade image generator in next update")
+                generated_images = []
+                for i in range(5):
+                    seed = random.randint(0,999999)
+                    image = client.text_to_image(f"{query}{message_text}", negative_prompt=f"{seed}")
+                    generated_images.append = gr.Image(image)
+                    yield generated_images
+                gr.Info("We are going to more upgrade image generator in next update")
             elif json_data["name"] == "image_qna":
                 inputs = llava(message, history)
                 streamer = TextIteratorStreamer(processor, skip_prompt=True, **{"skip_special_tokens": True})