Update app.py

app.py

@@ -26,7 +26,7 @@ def load_agent(model_id_1, model_id_2):
     # Load the video
     video_path_2 = hf_hub_download(model_id_2, filename="replay.mp4")
     
-    return video_path_1, results_1, video_path_2, results_2
+    return model_id_1, video_path_1, results_1, model_id_2, video_path_2, results_2
 
 def parse_metrics_accuracy(meta):
     if "model-index" not in meta:
@@ -60,9 +60,17 @@ gr.Interface(load_agent,
             label="Model 2",
         ),
     ],
-    ["video", gr.Textbox(
+    [ "text", "video", gr.Textbox(
             label="Mean Reward +/- Std Reward",
-        ), "video",  gr.Textbox(
+        ), "text", "video",  gr.Textbox(
             label="Mean Reward +/- Std Reward",
-        )]
+        )],
+        examples=[
+        ["sb3/a2c-AntBulletEnv-v0","sb3/ppo-AntBulletEnv-v0"],
+        ["ThomasSimonini/a2c-AntBulletEnv-v0", "sb3/a2c-AntBulletEnv-v0"],
+        ["sb3/dqn-SpaceInvadersNoFrameskip-v4", "sb3/a2c-SpaceInvadersNoFrameskip-v4"], 
+        ["ThomasSimonini/ppo-QbertNoFrameskip-v4","sb3/ppo-QbertNoFrameskip-v4"],
+    ],
+    title="Compare Deep Reinforcement Learning agents",
+    description="Type two models id you want to compare"
     ).launch()