Update app.py

app.py

@@ -98,8 +98,7 @@ def main():
                     #zeroshot = gr.Textbox(label="LaViLa (zero-shot) prediction")
                     #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.Image(label="Ego-VPA prediction")
-                    #ours = gr.Gallery(label="Ego-VPA prediction", columns=[3], rows=[1], object_fit="contain", height="auto")
+                    ours = gr.Gallery(label="Ego-VPA prediction", columns=[1], 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])