Create Dockerfile

Dockerfile

@@ -0,0 +1,25 @@
+FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 
+
+RUN apt-get update && apt-get install -y \
+    ffmpeg \
+    tk \
+    python3-tk \
+    && 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
+
+COPY . .
+
+EXPOSE 8000
+
+ENV TMP_DIRECTORY=tmp
+
+RUN mkdir -p ${TMP_DIRECTORY}
+
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]