try to use cuda

Dockerfile

@@ -1,10 +1,13 @@
 # Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
 # you will also find guides on how best to write your Dockerfile
 
-FROM python:3.10-slim
+FROM nvidia/cuda:12.1.0-cudnn8-runtime-ubuntu22.04
 
-# Install system dependencies
+# Install Python and system dependencies
 RUN apt-get update && apt-get install -y \
+    python3.10 \
+    python3.10-dev \
+    python3-pip \
     build-essential \
     git \
     curl \
@@ -25,7 +28,7 @@ ENV PATH="/home/user/.local/bin:$PATH"
 
 # Copy requirements first to leverage Docker cache
 COPY --chown=user requirements.txt .
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
+RUN pip3 install --no-cache-dir --upgrade -r requirements.txt
 
 # Copy application files
 COPY --chown=user *.py /app/
@@ -36,9 +39,12 @@ COPY --chown=user static /app/static/
 # Set environment variables
 ENV PYTHONPATH=/app
 ENV PYTHONUNBUFFERED=1
+ENV CUDA_HOME=/usr/local/cuda
+ENV PATH=${CUDA_HOME}/bin:${PATH}
+ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 
 # Expose the port the server runs on
 EXPOSE 7860
 
 # Run the server using main.py
-CMD ["python", "main.py", "--host", "0.0.0.0", "--port", "7860", "--model", "tiny", "--backend", "faster-whisper", "--task", "transcribe"]
+CMD ["python3", "main.py", "--host", "0.0.0.0", "--port", "7860", "--model", "base", "--backend", "faster-whisper", "--task", "transcribe"]