using CUDA image

Dockerfile

@@ -1,5 +1,9 @@
 # Dockerfile
-FROM tiangolo/uvicorn-gunicorn-fastapi:python3.9
+#FROM tiangolo/uvicorn-gunicorn-fastapi:python3.9
+FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
+
+ARG DEBIAN_FRONTEND=noninteractive
+ENV PYTHONUNBUFFERED=1
 
 # Set environment variables to prevent Python from writing pyc files to disk
 # and to force unbuffered output (useful for Docker)
@@ -14,10 +18,19 @@ RUN useradd -m -u 1000 user
 
 # Switch to the "user" user
 USER user
-
+# Set home to the user's home directory
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH \
+    PYTHONPATH=$HOME/app \
+	PYTHONUNBUFFERED=1 \
+	GRADIO_ALLOW_FLAGGING=never \
+	GRADIO_NUM_PORTS=1 \
+	GRADIO_SERVER_NAME=0.0.0.0 \
+	GRADIO_THEME=huggingface \
+	SYSTEM=spaces
 
 # Set the working directory in the container
-WORKDIR /app
+WORKDIR $HOME/app
 
 # Upgrade pip to the latest version and install Python dependencies
 COPY --chown=user requirements.txt .