Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import streamlit as st
 import numpy as np
 import pandas as pd
@@ -5,10 +6,20 @@ import matplotlib.pyplot as plt
 import seaborn as sns
 import cv2
 import torch
-from huggingface_hub import hf_hub_download, login
+from huggingface_hub import hf_hub_download
 from PIL import Image
 import pickle
 
+# Obtener el token de Hugging Face desde las variables de entorno
+hf_token = os.getenv("HF_API_TOKEN")
+
+if hf_token is None:
+    st.error("El token de Hugging Face no está configurado. Por favor, configura el secreto 'HF_API_TOKEN'.")
+    st.stop()
+
+# Autenticarse con el token
+from huggingface_hub import login
+login(token=hf_token)
 
 # Cargar dataset
 dataset_repo = "VicGerardoPR/Traffic_sign_dataset"