Adicionando a variavel de token

app.py

@@ -6,21 +6,25 @@ import pickle
 from PIL import Image
 import zipfile
 import os
+import os as _os
+
+# Pega o token do Hugging Face da variável de ambiente
+hf_token = _os.getenv("HUGGINGFACE_HUB_TOKEN")
 
 # Repositório no Hugging Face
 repo_id = "davebraga/wrdbTI6"
 
-# Baixar o modelo compactado
-model_zip_path = hf_hub_download(repo_id, "saved_model.zip")
+# Baixar o modelo compactado com autenticação
+model_zip_path = hf_hub_download(repo_id, "saved_model.zip", token=hf_token)
 
 # Descompactar o modelo se ainda não existir
 if not os.path.exists("saved_model"):
     with zipfile.ZipFile(model_zip_path, 'r') as zip_ref:
         zip_ref.extractall("saved_model")
 
-# Baixar os encoders
-category_encoder_path = hf_hub_download(repo_id, "category_encoder.pkl")
-color_encoder_path = hf_hub_download(repo_id, "color_encoder.pkl")
+# Baixar os encoders com autenticação
+category_encoder_path = hf_hub_download(repo_id, "category_encoder.pkl", token=hf_token)
+color_encoder_path = hf_hub_download(repo_id, "color_encoder.pkl", token=hf_token)
 
 # Carregar modelo e encoders
 model = load_model("saved_model")