Thomas Simonini
Add load video
92913ed
raw history blame
No virus
926 Bytes
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)