Update app.py

app.py

@@ -5,13 +5,13 @@ import torch
 
 app = FastAPI()
 
-# Définir un dossier cache accessible pour éviter les erreurs de permission
+# Définir un dossier cache accessible
 os.environ["TRANSFORMERS_CACHE"] = "/tmp"
 
 # Charger le modèle et le tokenizer depuis Hugging Face avec cache local
 MODEL_NAME = "fatmata/psybot"
-local_dir = "/app/model"  # Dossier pour stocker le modèle
-os.makedirs(local_dir, exist_ok=True)
+local_dir = "/tmp/model"  # Changer le dossier vers /tmp/model
+os.makedirs(local_dir, exist_ok=True)  # Crée le dossier si nécessaire
 
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, cache_dir=local_dir)
 model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, cache_dir=local_dir, torch_dtype=torch.float32)