Update app.py

app.py

@@ -5,7 +5,14 @@ def ai(text):
     dalle = replicate.models.get("borisdayma/dalle-mini")
     output = dalle.predict(prompt=text, n_predictions=4)
     print(output[0]["image"])
-    return output[0]["image"]
+    return [output[0]["image"], output[1]["image"]]
+
+with gr.Blocks() as demo:
+    inp = gr.Textbox(placeholder="Gandalf holding a staff that will open a portal to Middle Earth, digital art")
+    with gr.Row():
+        out = gr.Image()
+        out2 = gr.Image()
+    btn = gr.Button("Run")
+    btn.click(fn=update, inputs=inp, outputs=[out, out2])
 
-demo = gr.Interface(fn=ai, inputs="text", outputs="image")
 demo.launch()