Update app.py

app.py

@@ -3,22 +3,30 @@ import requests.exceptions
 from huggingface_hub import HfApi, hf_hub_download
 from huggingface_hub.repocard import metadata_load
 
-def load_agent(model_id):
+def load_agent(model_id_1, model_id_2):
     """
     This function load the agent's video and results
     :return: video_path
     """
     # Load the metrics
-    metadata = get_metadata(model_id)
+    metadata_1 = get_metadata(model_id_1)
     
     # Get the accuracy
-    results = parse_metrics_accuracy(metadata)
+    results_1 = parse_metrics_accuracy(metadata_1)
     
     # Load the video
-    video_path = hf_hub_download(model_id, filename="replay.mp4")
-
-    return video_path, results
-
+    video_path_1 = hf_hub_download(model_id_1, filename="replay.mp4")
+    
+    # Load the metrics
+    metadata_2 = get_metadata(model_id_2)
+    
+    # Get the accuracy
+    results_2 = parse_metrics_accuracy(metadata_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
 
 def parse_metrics_accuracy(meta):
     if "model-index" not in meta:
@@ -43,14 +51,18 @@ def get_metadata(model_id):
         return None
 
 
-agent1 = gr.Interface(load_agent, gr.Textbox(
-            label="model_1",
-        ), ["video", "text"])
-agent2 = gr.Interface(load_agent,  gr.Textbox(
-            label="model_2",
-        ), ["video", "text"])
-
-
-
-gr.Series(agent1, agent2).launch()
-
+gr.Interface(load_agent, 
+[
+        gr.Textbox(
+            label="Model 1",
+        ),
+        gr.Textbox(
+            label="Model 2",
+        ),
+    ],
+    ["video", gr.Textbox(
+            label="Mean Reward +/- Std Reward",
+        ), "video",  gr.Textbox(
+            label="Mean Reward +/- Std Reward",
+        )]
+    ).launch()