Update app.py

app.py

@@ -1,13 +1,20 @@
+import os
 from fastapi import FastAPI
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 app = FastAPI()
 
-# Charger le modèle et le tokenizer depuis Hugging Face
+# Définir un dossier cache accessible pour éviter les erreurs de permission
+os.environ["TRANSFORMERS_CACHE"] = "/tmp"
+
+# Charger le modèle et le tokenizer depuis Hugging Face avec cache local
 MODEL_NAME = "fatmata/psybot"
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float32)
+local_dir = "/app/model"  # Dossier pour stocker le modèle
+os.makedirs(local_dir, exist_ok=True)
+
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, cache_dir=local_dir)
+model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, cache_dir=local_dir, torch_dtype=torch.float32)
 
 @app.get("/")
 def home():