Update app.py

app.py

@@ -14,16 +14,20 @@ def open_ai_txt2img(prompt):
     return response["data"][0]["url"]
 
 # Gradio Interface
-def generator(prompt):
-    return open_ai_txt2img(prompt), open_ai_txt2img(prompt), open_ai_txt2img(prompt)
+# def generator(prompt):
+#     return open_ai_txt2img(prompt), open_ai_txt2img(prompt), open_ai_txt2img(prompt)
+
+# with gr.Blocks() as demo:
+#     prompt = gr.Textbox(label="Prompt")
+#     submit = gr.Button(label="Generate")
+#     image1 = gr.Image()
+#     image2 = gr.Image()
+#     image3 = gr.Image()
 
-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="mmd")
+# submit.click(generator, inputs=[prompt], outputs=[image1, image2, image3], api_name="mmd")
+
+def generator(prompt):
+    return open_ai_txt2img(prompt)
 
+demo = gr.Interface(fn=generator, inputs="text", outputs="image")
 demo.launch()