Update app.py

app.py

@@ -96,9 +96,10 @@ def main():
                 with gr.Column():
                     idx = gr.Number(label="Idx", visible=False)
                     #zeroshot = gr.Textbox(label="LaViLa (zero-shot) prediction")
-                    #zeroshot = gr.Gallery(label="LaViLa (zero-shot) prediction", columns=[2], rows=[1], object_fit="contain", height="auto")
+                    #zeroshot = gr.Gallery(label="LaViLa (zero-shot) prediction", columns=[3], rows=[1], object_fit="contain", height="auto")
                     #ours = gr.Textbox(label="Ego-VPA prediction")
-                    ours = gr.Gallery(label="Ego-VPA prediction", columns=[2], rows=[1], object_fit="contain", height="auto")
+                    ours = gr.Image(label="Ego-VPA prediction")
+                    #ours = gr.Gallery(label="Ego-VPA prediction", columns=[3], rows=[1], object_fit="contain", height="auto")
             btn = gr.Button("Predict", variant="primary")
             btn.click(predict_t2v, inputs=[idx], outputs=[ours])
             gr.Examples(examples=[[i, x] for i, x in enumerate(t2v_samples)], inputs=[idx, text])