Update app.py

app.py

@@ -11,9 +11,14 @@ app = FastAPI()
 BOT_USERNAME = "@DiscussionBot"
 MODEL = "gpt2"
 
-# Définir un répertoire de cache accessible
-CACHE_DIR = os.getenv("HF_HOME", "/app/cache")
-os.makedirs(CACHE_DIR, exist_ok=True)  # Créer le répertoire s'il n'existe pas
+# Définir un répertoire de cache accessible pour Hugging Face Spaces
+CACHE_DIR = os.getenv("HF_HOME", "/tmp/cache/huggingface")
+try:
+    os.makedirs(CACHE_DIR, exist_ok=True)  # Créer le répertoire s'il n'existe pas
+    print(f"Cache directory set to {CACHE_DIR}")
+except Exception as e:
+    print(f"Warning: Unable to create cache directory {CACHE_DIR}: {str(e)}")
+    raise
 
 # Vérification du jeton et connexion à Hugging Face
 try:
@@ -30,6 +35,7 @@ except Exception as e:
 # Initialisation du pipeline
 try:
     pipe = pipeline("text-generation", model=MODEL, token=HF_TOKEN if HF_TOKEN else None, cache_dir=CACHE_DIR)
+    print(f"Successfully initialized pipeline for model {MODEL}")
 except Exception as e:
     print(f"Failed to initialize pipeline for model {MODEL}: {str(e)}")
     raise