diff --git a/app/web/main.py b/app/web/main.py index 62a9738..c23f68b 100644 --- a/app/web/main.py +++ b/app/web/main.py @@ -16,12 +16,6 @@ from app.web.security import authenticate_api_key app = FastAPI() celery = get_celery_binding() -api_router = APIRouter( - prefix="/api/v1", - dependencies=[Depends(authenticate_api_key)], - responses={**DEFAULT_RESPONSES}, -) - def queue_task(job: models.Job) -> None: # queue an async transcription task. @@ -32,6 +26,13 @@ def queue_task(job: models.Job) -> None: transcribe.delay(job.id) +api_router = APIRouter( + prefix="/api/v1", + dependencies=[Depends(authenticate_api_key)], + responses={**DEFAULT_RESPONSES}, +) + + @api_router.get("/", response_model=None, status_code=204) def api_root() -> None: return None diff --git a/docker/prod/docker-compose.yml b/docker/prod/docker-compose.yml index 1afc165..77c2608 100644 --- a/docker/prod/docker-compose.yml +++ b/docker/prod/docker-compose.yml @@ -22,7 +22,7 @@ services: container_name: whisperbox_worker build: context: ../../ - # ENABLE GPU SUPPORT + # # dockerfile: docker/prod/worker.gpu.Dockerfile dockerfile: docker/prod/worker.Dockerfile environment: *app-variables @@ -33,7 +33,7 @@ services: - app depends_on: - redis - # ENABLE GPU SUPPORT + # # deploy: # resources: # reservations: @@ -55,7 +55,8 @@ services: networks: - app depends_on: - - worker + worker: + condition: service_healthy volumes: - whisperbox-data:/etc/whisperbox/data - whisper-models:/models diff --git a/docker/prod/worker.Dockerfile b/docker/prod/worker.Dockerfile index 73f8dc3..b0e61e6 100644 --- a/docker/prod/worker.Dockerfile +++ b/docker/prod/worker.Dockerfile @@ -24,7 +24,7 @@ COPY app ./app ENV VIRTUAL_ENV /opt/venv ENV PATH /opt/venv/bin:$PATH -# COPY scripts/download_model.py . -# RUN chmod +x download_model.py && python download_model.py ${WHISPER_MODEL:-small} +COPY scripts/download_model.py . +RUN chmod +x download_model.py && python download_model.py ${WHISPER_MODEL:-small} CMD celery --app=app.worker.main.celery worker --loglevel=info --concurrency=1