lumicollect / core /uploader.py
nemabruh404's picture
Create uploader.py
499c114 verified
Raw
History Blame
570 Bytes
import uuid
from huggingface_hub import HfApi
from config import DATASET_ID, HF_TOKEN
api = HfApi()
def upload_audio(audio_path: str, score: float) -> str:
"""
Upload file âm thanh lên HuggingFace Dataset.
Trả về filename nếu thành công, raise Exception nếu lỗi.
"""
filename = f"lumi_{score:.2f}_{uuid.uuid4().hex[:4]}.wav"
api.upload_file(
path_or_fileobj=audio_path,
path_in_repo=f"audios/{filename}",
repo_id=DATASET_ID,
repo_type="dataset",
token=HF_TOKEN,
)
return filename