Changed app image

Dockerfile

@@ -1,7 +1,7 @@
 # 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.9
+FROM nvidia/cuda:12.5.1-cudnn-devel-ubuntu20.04
 
 RUN useradd -m -u 1000 user
 USER user
@@ -10,7 +10,6 @@ ENV PATH="/home/user/.local/bin:$PATH"
 WORKDIR /app
 
 COPY --chown=user ./requirements.txt requirements.txt
-RUN apt-get update && apt-get install libgl1
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 COPY --chown=user . /app