Update Dockerfile

Dockerfile

@@ -1,5 +1,19 @@
-FROM python:3.10.9
+ARG BASE=nvidia/cuda:11.8.0-base-ubuntu22.04
+FROM ${BASE}
 
+# Install OS dependencies:
+RUN apt-get update && apt-get upgrade -y \
+    && apt-get install -y --no-install-recommends \
+    gcc g++ \
+    make \
+    python3 python3-dev python3-pip python3-venv python3-wheel \
+    espeak-ng libsndfile1-dev \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install Major Python Dependencies:
+RUN pip3 install llvmlite --ignore-installed \
+    && pip3 install torch torchaudio --extra-index-url https://download.pytorch.org/whl/cu118 \
+    && rm -rf /root/.cache/pip
 
 # Set the working directory
 WORKDIR /app
@@ -8,7 +22,7 @@ WORKDIR /app
 COPY . .
 
 # Install Python dependencies
-RUN pip install --no-cache-dir  --user -r requirements.txt
-
+RUN pip install --no-cache-dir --user -r requirements.txt
 
+# Command to run the application
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]