fix: compose

Author: leoguillaume

README.md

@@ -1,3 +1,9 @@
-# whisper-openai-api
+# Whisper OpenAI API
 
 
+## Environment variables
+
+| Variable | Description |
+| -------- | ----------- |
+| MODELS_CACHE_DIR | Directory to store the models |
+| API_KEY | API key for API access (optional) |

app/Dockerfile

@@ -1,8 +1,13 @@
-FROM python:3.12-slim
+FROM nvidia/cuda:12.5.1-cudnn-devel-ubuntu22.04
 
-RUN apt-get update && \
-    apt-get install -y python3-pip python3-dev curl && \
-    rm -rf /var/lib/apt/lists/*
+ENV PYTHONUNBUFFERED=1
+ENV DEBIAN_FRONTEND=noninteractive
+ENV TZ=Etc/UTC
+
+RUN apt-get update -y && \
+  apt-get -y upgrade && \
+  apt-get install -y ffmpeg python3 python3-pip python3-dev curl && \
+  rm -rf /var/lib/apt/lists/*
 
 RUN groupadd --gid 1100 whisper
 RUN useradd --home /home/whisper --gid 1100 --uid 1100 whisper 

compose.yml

@@ -1,6 +1,6 @@
 services:  
   app:
-    image: ghcr.io/etalab-ia/whisper-openai-api:latest
+    image: ghcr.io/etalab-ia/whisper-openai-api/app:latest
     restart: always
     command: python3 app/main.py --model openai/whisper-large-v3
     ports: