Update Dockerfile

Dockerfile

@@ -6,14 +6,11 @@ RUN apt-get update && apt-get install -y \
     ffmpeg \
     tk \
     python3-tk \
+    python3-pip \
     && rm -rf /var/lib/apt/lists/*
 
-RUN python3 -m pip install --upgrade pip
-
-RUN pip install onnxruntime-gpu
-
 RUN --mount=type=cache,target=/root/.cache \
-    pip3 install "audio-separator" fastapi uvicorn python-multipart ffmpeg-python pydantic youtube-transcript-api
+    pip3 install "audio-separator[gpu]" fastapi uvicorn python-multipart ffmpeg-python pydantic youtube-transcript-api
 
 COPY . .