Update app.py

app.py

@@ -241,10 +241,44 @@ def create_demo(process, max_images=12, default_num_images=4):
             gr.Markdown(description)
 
         with gr.Row():
-            gr.Markdown("We generated a trajectory of our simulated environment. We will then use it with our different models.")
-            gr.Video("./trajectory_hf/trajectory.avi",
-                    format="avi",
-                    interactive=False)
+            with gr.Column():
+                gr.Markdown("We generated a trajectory of our simulated environment. We will then use it with our different models.")
+            with gr.Column():
+                gr.Video("./trajectory_hf/trajectory.avi",
+                        format="avi",
+                        interactive=False)
+
+        with gr.Row():
+            with gr.Column():
+                gr.Markdown("Trajectory processed with coyo-500k model :")
+            with gr.Column():
+                gr.Video("./trajectory_hf/trajectory_coyo-500k.avi",
+                        format="avi",
+                        interactive=False)
+
+        with gr.Row():
+            with gr.Column():
+                gr.Markdown("Trajectory processed with bridge-2M model :")
+            with gr.Column():
+                gr.Video("./trajectory_hf/trajectory_bridge-2M.avi",
+                        format="avi",
+                        interactive=False)
+
+        with gr.Row():
+            with gr.Column():
+                gr.Markdown("Trajectory processed with coyo1M-bridge2M model :")
+            with gr.Column():
+                gr.Video("./trajectory_hf/trajectory_coyo1M-bridge2M.avi",
+                        format="avi",
+                        interactive=False)
+
+        with gr.Row():
+            with gr.Column():
+                gr.Markdown("Trajectory processed with coyo28-bridge4 model :")
+            with gr.Column():
+                gr.Video("./trajectory_hf/trajectory_coyo28-bridge4.avi",
+                        format="avi",
+                        interactive=False)