Update main.py

main.py

@@ -21,6 +21,12 @@ from pptx import Presentation
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Configuration du cache pour Hugging Face
+TEMP_CACHE_DIR = "/tmp/huggingface_cache"
+os.environ["TRANSFORMERS_CACHE"] = TEMP_CACHE_DIR
+os.environ["HF_HOME"] = TEMP_CACHE_DIR
+Path(TEMP_CACHE_DIR).mkdir(parents=True, exist_ok=True)
+
 # Initialisation de l'application FastAPI
 app = FastAPI()
 
@@ -38,11 +44,6 @@ BASE_DIR = Path(__file__).parent
 UPLOAD_FOLDER = BASE_DIR / "uploads"
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
-# Configuration du cache pour Hugging Face
-TEMP_CACHE_DIR = "/tmp/huggingface_cache"
-os.environ["TRANSFORMERS_CACHE"] = TEMP_CACHE_DIR
-os.environ["HF_HOME"] = TEMP_CACHE_DIR
-
 # Modèles Hugging Face
 HF_TOKEN = os.getenv("HF_TOKEN")
 client = InferenceClient(token=HF_TOKEN)