diff --git a/effi-stt/docker-compose.yml b/effi-stt/docker-compose.yml index ab556a8..6f43fe8 100644 --- a/effi-stt/docker-compose.yml +++ b/effi-stt/docker-compose.yml @@ -6,11 +6,11 @@ services: ports: - "8200:8200" environment: - - MODEL_SIZE=large-v3 + - MODEL_SIZE=distil-large-v3 - DEVICE=cpu - COMPUTE_TYPE=int8 - LANGUAGE=fr - - CPU_THREADS=4 + - CPU_THREADS=8 - INACTIVITY_TIMEOUT=300 volumes: # Cache pour les modèles (évite de re-télécharger) diff --git a/effi-stt/main.py b/effi-stt/main.py index 08ed82e..adcaeff 100644 --- a/effi-stt/main.py +++ b/effi-stt/main.py @@ -22,7 +22,7 @@ app = FastAPI( version="1.1.0" ) -MODEL_SIZE = os.getenv("MODEL_SIZE", "large-v3") +MODEL_SIZE = os.getenv("MODEL_SIZE", "distil-large-v3") DEVICE = os.getenv("DEVICE", "cpu") COMPUTE_TYPE = os.getenv("COMPUTE_TYPE", "int8") LANGUAGE = os.getenv("LANGUAGE", "fr") @@ -46,7 +46,7 @@ def load_model(): MODEL_SIZE, device=DEVICE, compute_type=COMPUTE_TYPE, - cpu_threads=int(os.getenv("CPU_THREADS", "4")) + cpu_threads=int(os.getenv("CPU_THREADS", "8")) ) model_loaded = True logger.info("✅ Modèle chargé avec succès!")