Update app.py

app.py

@@ -1,3 +1,8 @@
+import os
+
+# Imposta la cache dei modelli in una cartella scrivibile
+os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
+
 from fastapi import FastAPI, Request
 from fastapi.responses import JSONResponse, FileResponse
 from transformers import AutoModelForCausalLM, AutoTokenizer
@@ -5,6 +10,9 @@ import uvicorn
 
 app = FastAPI()
 
+# Crea la cartella di cache se non esiste
+os.makedirs("/app/cache", exist_ok=True)
+
 # Carica il modello Hugging Face
 model_name = "microsoft/DialoGPT-small"
 tokenizer = AutoTokenizer.from_pretrained(model_name)