app.py

import gradio as gr
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):
 """
 This function load the agent's video and results
 :return: video_path
 """
 # Load the metrics
 metadata_1 = get_metadata(model_id_1)
 
 # Get the accuracy
 results_1 = parse_metrics_accuracy(metadata_1)
 
 # Load the video
 video_path_1 = hf_hub_download(model_id_1, filename="replay.mp4")
 
 return model_id_1, video_path_1, results_1

def parse_metrics_accuracy(meta):
 if "model-index" not in meta:
 return None
 result = meta["model-index"][0]["results"]
 metrics = result[0]["metrics"]
 accuracy = metrics[0]["value"]
 return accuracy

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




with app:
 gr.Markdown(
 """
 # Observa los agentes de Lagomorph en acción 🤖
 
 Selecciona el modelo a observar.
 """)
 with gr.Row():
 model1_input = gr.Dropdown(["odiaz1066/CartPoleGoal-Lagomorph-seed42",
 "odiaz1066/BreakoutNoFrameskip-v4-dqn_atari-seed1",
 "odiaz1066/PongNoFrameskip-v4-dqn_atari-seed1",
 "odiaz1066/Reinforce-CartPole-v1",
 "odiaz1066/DQN-SpaceInvadersNoFrameskip-v4",
 "odiaz1066/PPO-LunarLander",
 "odiaz1066/q-FrozenLake-v1-4x4-noSlippery",
 "odiaz1066/q-taxi-v3",
 "odiaz1066/Reinforce-Pixelcopter-PLE-v0",
 "odiaz1066/a2c-AntBulletEnv-v0",
 "odiaz1066/a2c-PandaReachDense-v2"],
 label="Modelo")
 with gr.Row():
 app_button = gr.Button("Mostrar modelo")
 with gr.Row():
 with gr.Column():
 model1_name = gr.Markdown()
 model1_video_output = gr.Video()
 model1_score_output = gr.Textbox(label="Recompensa +/- Variación")

 app_button.click(load_agent, inputs=[model1_input], outputs=[model1_name, model1_video_output, model1_score_output])
 
 examples = gr.Examples(examples=[["odiaz1066/CartPoleGoal-Lagomorph-seed42"],
 ["odiaz1066/BreakoutNoFrameskip-v4-dqn_atari-seed1"],
 ["odiaz1066/PongNoFrameskip-v4-dqn_atari-seed1"],
 ["odiaz1066/Reinforce-CartPole-v1"],
 ["odiaz1066/DQN-SpaceInvadersNoFrameskip-v4"],
 ["odiaz1066/PPO-LunarLander"],
 ["odiaz1066/q-FrozenLake-v1-4x4-noSlippery"],
 ["odiaz1066/q-taxi-v3"],
 ["odiaz1066/Reinforce-Pixelcopter-PLE-v0"],
 ["odiaz1066/a2c-AntBulletEnv-v0"],
 ["odiaz1066/a2c-PandaReachDense-v2"]],
 inputs=[model1_input])

 
app.launch()