test_model / hub_basics.py
Barzabel777's picture
add some things
4bf4171
from huggingface_hub import HfApi, create_repo, HfFileSystem
from dotenv import load_dotenv
import os
load_dotenv('.env')
HF_TOKEN = os.getenv('TOKEN')
REPO_NAME = "Barzabel777/test_model2"
# Создаем репозиторий через API
create_repo(
repo_id=REPO_NAME,
token=HF_TOKEN,
exist_ok=True, # Если True, не будет ошибки при существующем репозитории
repo_type="model", # Может быть: model, dataset, space
)
# Инициализируем API
api = HfApi(token=HF_TOKEN)
# Загружаем README.md
api.upload_file(
path_or_fileobj="data/README.md", # Локальный путь к файлу
path_in_repo="README.md", # Путь в репозитории
repo_id=REPO_NAME,
repo_type="model",
)
# Загружаем config.json
api.upload_file(
path_or_fileobj="data/test_data.json",
path_in_repo="config.json",
repo_id=REPO_NAME,
repo_type="model",
)
print(f"Репозиторий создан: https://huggingface.co/{REPO_NAME}")
fs = HfFileSystem()
repo_path = "Barzabel777/test_model2"
files = fs.ls(repo_path, detail=False)
print("Список файлов:")
for file in files:
print(file)