Update Dockerfile

Dockerfile

@@ -1,11 +1,24 @@
-# Use a PyTorch base image
-FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+# Use official CUDA base image
+FROM nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu20.04
 
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PYTHON_VERSION=3.10
+
+# Install Python, pip, and dependencies
+RUN apt-get update && apt-get install -y \
+    python3.10 python3.10-dev python3.10-distutils \
+    curl git ffmpeg libsm6 libxext6 wget build-essential \
+    && ln -s /usr/bin/python3.10 /usr/bin/python \
+    && curl -sS https://bootstrap.pypa.io/get-pip.py | python
+
+# Set working directory
 WORKDIR /app
 
+# Install Python dependencies
 COPY requirements.txt .
 RUN pip install --upgrade pip && pip install -r requirements.txt
 
+# Copy app code
 COPY app.py .
 
 EXPOSE 7860