Update app.py

app.py

@@ -3,6 +3,8 @@ import requests.exceptions
 from huggingface_hub import HfApi, hf_hub_download
 from huggingface_hub.repocard import metadata_load
 
+app = gr.Blocks()
+
 def load_agent(model_id_1, model_id_2):
     """
     This function load the agent's video and results
@@ -51,26 +53,42 @@ def get_metadata(model_id):
         return None
 
 
-gr.Interface(load_agent, 
-[
-        gr.Textbox(
-            label="Model 1",
-        ),
-        gr.Textbox(
-            label="Model 2",
-        ),
-    ],
-    [ "text", "video", gr.Textbox(
-            label="Mean Reward +/- Std Reward",
-        ), "text", "video",  gr.Textbox(
-            label="Mean Reward +/- Std Reward",
-        )],
+
+
+with app:
+    gr.Markdown(
+    """
+    # Compare Deep Reinforcement Learning Agents 🤖
+    
+    Type two models id you want to compare or check examples below.
+    """)
+    with gr.Row():
+      model1_input = gr.Textbox(label="Model 1")
+      model2_input = gr.Textbox(label="Model 2")
+    with gr.Row():
+      app_button = gr.Button("Compare models")
+    with gr.Row():
+      with gr.Column():
+        model1_name = gr.Markdown()
+        model1_video_output = gr.Video()
+        model1_score_output = gr.Textbox(label="Mean Reward +/- Std Reward")
+      with gr.Column():
+        model2_name = gr.Markdown()
+        model2_video_output = gr.Video()
+        model2_score_output = gr.Textbox(label="Mean Reward +/- Std Reward")
+
+    app_button.click(load_agent, inputs=[model1_input, model2_input], outputs=[model1_name,  model1_video_output, model1_score_output, model2_name, model2_video_output, model2_score_output])
+    
+app.launch()
+
+
+
+
+"""
+
         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 or check examples below."
-    ).launch()
+"""