Update Dockerfile

Dockerfile

@@ -1,15 +1,9 @@
-FROM nvidia/cuda:12.1.1-base-ubi8
+FROM python:3.9
 WORKDIR /app
 
 COPY ./requirements.txt /app/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /app/requirements.txt
 
-ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH \
-    D2M_PORT=7860 \
-    GRADIO_PORT=8000 \
-    URL_GRADIO=https://localhost:8000/
-
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 
@@ -23,6 +17,12 @@ ENV HOME=/home/user \
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app
 
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH \
+    D2M_PORT=7860 \
+    GRADIO_PORT=8000 \
+    URL_GRADIO=https://localhost:8000/
+
 COPY . .
 
 CMD ["python",  "main.py"]