Update app.py

app.py

@@ -15,7 +15,7 @@ def load_agent(model_id_1, model_id_2):
     results_1 = parse_metrics_accuracy(metadata_1)
     
     # Load the video
-    video_path_1 = hf_hub_download(model_id, filename="replay.mp4")
+    video_path_1 = hf_hub_download(model_id_1, filename="replay.mp4")
     
     # Load the metrics
     metadata_2 = get_metadata(model_id_2)
@@ -24,7 +24,7 @@ def load_agent(model_id_1, model_id_2):
     results_2 = parse_metrics_accuracy(metadata_2)
     
     # Load the video
-    video_path_2 = hf_hub_download(model_id, filename="replay.mp4")
+    video_path_2 = hf_hub_download(model_id_2, filename="replay.mp4")
     
     return video_path_1, results_1, video_path_2, results_2