Update Dockerfile

Dockerfile

@@ -1,14 +1,9 @@
-# Use the official Python 3.10.9 image
-FROM ubuntu
-# RUN adduser
-RUN useradd -m -s /bin/bash xyz
-RUN mkdir -p /usr/src/app/
-RUN chown -R xyz:xyz /usr/src/app/
+FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
 
 WORKDIR /usr/src/app
-# RUN add-apt-repository main
+
 RUN apt-get update 
-# RUN apt-get install -y libgl1-mesa-glx
+
 RUN apt install -y python3-pip 
 RUN rm /usr/lib/python3.*/EXTERNALLY-MANAGED
 
@@ -23,6 +18,4 @@ ENV HF_HOME=/tmp
 ENV TRANSFORMERS_CACHE=/tmp
 
 
-
-# Start the FastAPI app on port 7860, the default port expected by Spaces
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]