felipecaspol commited on
Commit
be48726
·
1 Parent(s): b26704f

Corrigiendo API para modelo fusionado

Browse files
Files changed (1) hide show
  1. api.py +3 -2
api.py CHANGED
@@ -11,8 +11,8 @@ app = FastAPI()
11
  # ✅ Definir un directorio de caché seguro
12
  os.environ["HF_HOME"] = "/tmp/huggingface"
13
 
14
- # ✅ Nombre del modelo en Hugging Face Hub (modelo fusionado)
15
- HUGGING_FACE_REPO = "fcp2207/Fusion_modelo_Phi2" # Ruta donde subiste el modelo fusionado
16
 
17
  # ✅ Descargar el modelo fusionado desde Hugging Face
18
  print("🔄 Descargando modelo fusionado...")
@@ -67,3 +67,4 @@ if __name__ == "__main__":
67
 
68
 
69
 
 
 
11
  # ✅ Definir un directorio de caché seguro
12
  os.environ["HF_HOME"] = "/tmp/huggingface"
13
 
14
+ # ✅ Repositorio del modelo fusionado (actualizado)
15
+ HUGGING_FACE_REPO = "fcp2207/Fusion_modelo_Phi2" # Debe coincidir con el Space donde guardaste `phi2_full_model`
16
 
17
  # ✅ Descargar el modelo fusionado desde Hugging Face
18
  print("🔄 Descargando modelo fusionado...")
 
67
 
68
 
69
 
70
+