Fixing GPT NEO app.py

app.py

@@ -1,6 +1,12 @@
+import os
 from fastapi import FastAPI
 from transformers import pipeline
 
+# Configurar el directorio de caché de Hugging Face
+os.makedirs('/app/cache', exist_ok=True)
+
+os.environ['TRANSFORMERS_CACHE'] = '/app/cache'
+
 app = FastAPI()
 
 # Cargar el modelo de GPT-Neo
@@ -9,4 +15,4 @@ generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
 @app.post("/generate")
 async def generate(prompt: str):
     response = generator(prompt, max_length=100)
-    return {"text": response[0]["generated_text"]}
+    return {"text": response[0]["generated_text"]}