Update arquivos.py

arquivos.py

@@ -1,8 +1,6 @@
 import os
-import shutil
 from huggingface_hub import HfApi, HfFolder, upload_file
 
-# Nome do Dataset Hugging Face
 DATASET_REPO = "pcdoido2/video-gerados"
 CATEGORIES = ["AVATAR WORLD", "BLOX FRUITS", "TOCA LIFE"]
 
@@ -18,13 +16,14 @@ def salvar_no_dataset(video_path, categoria):
 
     api = HfApi()
 
-    # Usa o token do usuário (precisa estar configurado no Space ou local com HfFolder)
-    token = HfFolder.get_token()
+    # Usa o token do usuário (precisa estar configurado no Space ou no ambiente)
+    token = os.getenv("HUGGINGFACEHUB_API_TOKEN") or HfFolder.get_token()
 
-    if token is None:
-        raise ValueError("Token do Hugging Face não encontrado. Configure com HfFolder ou no ambiente.")
+    if not token:
+        raise ValueError(
+            "Token do Hugging Face não encontrado. Configure a variável HUGGINGFACEHUB_API_TOKEN nos Settings > Variables do Space."
+        )
 
-    # Faz upload do arquivo
     print(f"Fazendo upload de {video_path} para o dataset {DATASET_REPO} em {path_in_dataset}")
     upload_file(
         path_or_fileobj=video_path,