LitRL-Inference / src /huggingface /huggingface_client.py
c-gohlke's picture
Upload folder using huggingface_hub
5b344d4 verified
raw
history blame contribute delete
367 Bytes
from huggingface_hub import HfApi
from src.huggingface.get_environments import get_environments
from src.huggingface.get_files import get_mp4_paths
class HuggingFaceClient:
def __init__(self) -> None:
self.hf_api = HfApi()
self.environments = get_environments(self.hf_api)
self.mp4_paths = get_mp4_paths(environments=self.environments)