pippo

src/streamlit_app.py

@@ -20,6 +20,9 @@ DATA_DIR = "/app" #"/tmp"
 
 # 📌 Autenticazione Hugging Face dal Secret nello Space
 HF_TOKEN = os.getenv("HF_TOKEN")
+os.environ["HF_HOME"] = "/app/.cache"
+os.environ["HF_DATASETS_CACHE"] = "/app/.cache"
+
 
 if HF_TOKEN:
     api = HfApi()
@@ -44,7 +47,7 @@ dataset = load_dataset("sradc/imagenet_resized_64x64", split="train")
 image_list = []
 label_list = []
 
-#for i, (image, label) in enumerate(imagenet.take(300)):  # Prende solo 300 immagini  PER TENSORFLOW
+#for i, (image, label) in enumerate(imagenet.take(300)):  # Prende solo 300 immagini  PER TENSORFLOW 
 for i, sample in enumerate(dataset):
     if i >= 300:  # Prende solo 300 immagini
         break