proper use of api

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 import numpy as np
 import pandas as pd
 from apscheduler.schedulers.background import BackgroundScheduler
-from huggingface_hub import HfApi, hf_hub_download
+from huggingface_hub import HfApi
 
 from src.backend import backend_routine
 
@@ -127,7 +127,7 @@ def get_leaderboard_df():
 
     data = []
     for filename in filenames:
-        path = hf_hub_download(repo_id=RESULTS_REPO, filename=filename, repo_type="dataset")
+        path = API.hf_hub_download(repo_id=RESULTS_REPO, filename=filename, repo_type="dataset")
         with open(path) as fp:
             report = json.load(fp)
         user_id, model_id = report["config"]["model_id"].split("/")

src/backend.py

@@ -3,7 +3,7 @@ import os
 import re
 import tempfile
 
-from huggingface_hub import CommitOperationAdd, HfApi, hf_hub_download
+from huggingface_hub import CommitOperationAdd, HfApi
 
 from src.evaluation import evaluate
 from src.logging import setup_logger
@@ -33,7 +33,7 @@ def _backend_routine():
 
     evaluated_models = set()
     for filename in filenames:
-        path = hf_hub_download(repo_id=RESULTS_REPO, filename=filename, repo_type="dataset")
+        path = API.hf_hub_download(repo_id=RESULTS_REPO, filename=filename, repo_type="dataset")
         with open(path) as fp:
             report = json.load(fp)
         evaluated_models.add((report["config"]["model_id"], report["config"]["model_sha"]))

src/evaluation.py

@@ -6,7 +6,7 @@ import gymnasium as gym
 import numpy as np
 import torch
 from gymnasium import wrappers
-from huggingface_hub import HfApi, hf_hub_download
+from huggingface_hub import HfApi
 from huggingface_hub.utils._errors import EntryNotFoundError
 
 from src.logging import setup_logger
@@ -236,7 +236,7 @@ def evaluate(model_id, revision):
 
     # Check if the agent exists
     try:
-        agent_path = hf_hub_download(repo_id=model_id, filename="agent.pt")
+        agent_path = API.hf_hub_download(repo_id=model_id, filename="agent.pt")
     except EntryNotFoundError:
         logger.error("Agent not found")
         return None