Update app.py

app.py

@@ -41,8 +41,18 @@ def stable_diffusion_txt2img(prompt, api_key, model_key, model_inputs):
 
 # Gradio Interface
 def generator(prompt):
-  generated_image = stable_diffusion_txt2img(prompt, api_key, model_key, model_inputs)
-  return generated_image
+  generated_image1 = stable_diffusion_txt2img(prompt, api_key, model_key, model_inputs)
+  generated_image2 = stable_diffusion_txt2img(prompt, api_key, model_key, model_inputs)
+  generated_image3 = stable_diffusion_txt2img(prompt, api_key, model_key, model_inputs)
+  return generated_image1, generated_image2, generated_image3
 
-iface = gr.Interface(fn=generator, inputs="text", outputs="image")
-iface.launch()
+with gr.Blocks() as demo:
+  prompt = gr.Textbox(label="Prompt")
+  submit = gr.Button(label="Generate")
+  image1 = gr.Image()
+  image2 = gr.Image()
+  image3 = gr.Image()
+
+  submit.click(generator, inputs=[prompt], outputs=[image1, image2, image3], api_name="mmsd")
+
+demo.launch()