Add load video

.idea/.gitignore

@@ -0,0 +1,3 @@
+# Default ignored files
+/shelf/
+/workspace.xml

.idea/Compare-Reinforcement-Learning-Agents.iml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml

@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.8" project-jdk-type="Python SDK" />
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/Compare-Reinforcement-Learning-Agents.iml" filepath="$PROJECT_DIR$/.idea/Compare-Reinforcement-Learning-Agents.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+  </component>
+</project>

app.py

@@ -0,0 +1,32 @@
+import gradio as gr
+import requests.exceptions
+from huggingface_hub import HfApi, hf_hub_download
+from huggingface_hub.repocard import metadata_load
+
+def load_agent_video(model_id):
+    """
+    This function load the agent's video and results
+    :return: video_path
+    """
+    video_path = hf_hub_download(model_id, filename="replay.mp4")
+    return video_path
+
+def get_metadata(model_id):
+    """
+    Get the metadata of the model repo
+    :param model_id:
+    :return: metadata
+    """
+    try:
+        readme_path = hf_hub_download(model_id, filename="README.md")
+        metadata = metadata_load(readme_path)
+        print(metadata)
+        return metadata
+    except requests.exceptions.HTTPError:
+        return None
+
+agent1 = gr.Interface("text", load_agent_video(input), gr.Video(), "playable_video")
+agent2 = gr.Interface("text", load_agent_video(input), gr.Video(), "playable_video")
+
+gr.Parallel(agent1, agent2)
+