Update app.py

app.py

@@ -62,7 +62,8 @@ with gr.Blocks() as demo:
             height = gr.inputs.Slider(label="Height", default=256, step=32, maximum = 256, minimum=32)
             images = gr.inputs.Slider(label="Images - How many images you wish to generate", default=1, step=1, minimum=1, maximum=4)
             diversity = gr.inputs.Slider(label="Diversity scale - How different from one another you wish the images to be",default=15.0, minimum=1.0, maximum=15.0)
-            img = [gr.outputs.Image(type="pil"),gr.outputs.Textbox(label="Error")]
+            #gallery = [gr.outputs.Image(type="pil"),gr.outputs.Textbox(label="Error")]
+            gallery = gr.Gallery(label="Individual images")
             with gr.Row():
                 get_image_latent = gr.Button("Generate Image go brr")         
         with gr.Column():
@@ -73,7 +74,7 @@ with gr.Blocks() as demo:
 
         
     speech_to_text.click(speech_to_text, inputs=audio_file, outputs=text)
-    get_image_latent.click(text2image_latent, inputs=[text,steps,width,height,images,diversity], outputs=img)
+    get_image_latent.click(text2image_latent, inputs=[text,steps,width,height,images,diversity], outputs=gallery)
     zero_shot_clf.click(zero_shot, inputs=[img[0],text_input], outputs=label)
 
 demo.launch(enable_queue=False)