diff --git a/effi-stt/docker-compose.yml b/effi-stt/docker-compose.yml index 6f43fe8..7b542d6 100644 --- a/effi-stt/docker-compose.yml +++ b/effi-stt/docker-compose.yml @@ -6,7 +6,7 @@ services: ports: - "8200:8200" environment: - - MODEL_SIZE=distil-large-v3 + - MODEL_SIZE=large-v3 - DEVICE=cpu - COMPUTE_TYPE=int8 - LANGUAGE=fr diff --git a/effi-stt/main.py b/effi-stt/main.py index 91af4ff..1d7d098 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", "distil-large-v3") +MODEL_SIZE = os.getenv("MODEL_SIZE", "large-v3") DEVICE = os.getenv("DEVICE", "cpu") COMPUTE_TYPE = os.getenv("COMPUTE_TYPE", "int8") LANGUAGE = os.getenv("LANGUAGE", "fr")