update to UI

app.py

@@ -66,14 +66,16 @@ def main():
             """
         )
         with gr.Row():        
-            with gr.Column():
-                videos = [gr.Video(label=labels[i], format='mp4', height=256, min_width=340) for i in range(num_samples)]
-            with gr.Column():
+            videos = [
+                gr.Video(label=labels[i], format='mp4', height=256, width=256, autoplay=True) 
+                for i in range(num_samples)
+            ]
+            with gr.Row():
                 text = gr.Text(label="Query", visible=False)
                 label = gr.Text(label="Ground Truth")
                 ours = gr.Text(label="SViTT-Ego prediction")
-        btn = gr.Button("Predict", variant="primary")
-        btn.click(predict, inputs=[text], outputs=[label, ours])
+                btn = gr.Button("Predict", variant="primary")
+                btn.click(predict, inputs=[text], outputs=[label, ours])
         inputs = [text]
         inputs.extend(videos)
         gr.Examples(examples=[[sample_text[i], x[0], x[1], x[2], x[3], x[4]] for i, x in enumerate(sample_videos)], inputs=inputs)