KingNish commited on
Commit
68c5939
·
verified ·
1 Parent(s): b3c077f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -151,12 +151,10 @@ def respond(message, history):
151
  number_of_image = json_data["arguments"]["number_of_image"]
152
  gr.Info("Generating Image, Please wait 10 sec...")
153
  client = InferenceClient("stabilityai/stable-diffusion-3-medium-diffusers")
154
- generated_images = []
155
- for i in range(5):
156
- seed = random.randint(0,999999)
157
- image = client.text_to_image(f"{query}{message_text}", negative_prompt=f"{seed}")
158
- generated_images.append = gr.Image(image)
159
- yield generated_images
160
  gr.Info("We are going to more upgrade image generator in next update")
161
  elif json_data["name"] == "image_qna":
162
  inputs = llava(message, history)
 
151
  number_of_image = json_data["arguments"]["number_of_image"]
152
  gr.Info("Generating Image, Please wait 10 sec...")
153
  client = InferenceClient("stabilityai/stable-diffusion-3-medium-diffusers")
154
+ seed = random.randint(0,999999)
155
+ image = client.text_to_image(f"{query}{message_text}", negative_prompt=f"{seed}")
156
+ generated_images = gr.Image(image)
157
+ yield generated_images
 
 
158
  gr.Info("We are going to more upgrade image generator in next update")
159
  elif json_data["name"] == "image_qna":
160
  inputs = llava(message, history)