Corregido error de permisos en Hugging Face (caché personalizada)

api.py

@@ -1,26 +1,31 @@
-# Actualización forzada para subir a Hugging Face..
 from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 from huggingface_hub import hf_hub_download
+import os
 
 # ✅ Inicializar FastAPI
 app = FastAPI()
 
+# ✅ Crear directorio de caché para evitar problemas de permisos
+CACHE_DIR = "./cache"
+os.makedirs(CACHE_DIR, exist_ok=True)
+
 # ✅ Nombre del modelo en Hugging Face Hub
 HUGGING_FACE_REPO = "fcp2207/Phi-2"  # Asegúrate de que sea el nombre correcto en Hugging Face
 MODEL_FILENAME = "phi2_finetuned.pth"  # Nombre del archivo en Hugging Face
 
-# ✅ Descargar el modelo desde Hugging Face (sin autenticación)
+# ✅ Descargar el modelo desde Hugging Face (especificando caché)
 model_path = hf_hub_download(
     repo_id=HUGGING_FACE_REPO,
-    filename=MODEL_FILENAME
+    filename=MODEL_FILENAME,
+    cache_dir=CACHE_DIR  # Ruta de caché permitida
 )
 
 # ✅ Cargar el tokenizer y el modelo base
-tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2")
-model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2")
+tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2", cache_dir=CACHE_DIR)
+model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", cache_dir=CACHE_DIR)
 
 # ✅ Cargar los pesos del modelo
 model.load_state_dict(torch.load(model_path, map_location="cpu"))
@@ -60,3 +65,6 @@ if __name__ == "__main__":
 
 
 
+
+
+