Update Dockerfile

Dockerfile

@@ -2,12 +2,6 @@ FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
 ENV DEBIAN_FRONTEND=noninteractive
 
-# Set CUDA_HOME environment variable
-ENV CUDA_HOME=/usr/local/cuda
-ENV PATH=${CUDA_HOME}/bin:/home/user/.local/bin:${PATH}
-ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
-ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH}
-
 # Set the MKL_THREADING_LAYER environment variable to GNU
 ENV MKL_THREADING_LAYER=GNU
 
@@ -25,9 +19,16 @@ 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
+	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 \
+    GRADIO_SHARE=False \
+	SYSTEM=spaces
 
 # Set the environment variable to specify the GPU device
 ENV CUDA_DEVICE_ORDER=PCI_BUS_ID