Update Dockerfile

Dockerfile

@@ -29,22 +29,25 @@ RUN update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
 # Switch to non-root user
 USER user
 
-# Create virtual environment
-RUN python -m venv /home/user/venv
-ENV VIRTUAL_ENV=/home/user/venv
-ENV PATH="$VIRTUAL_ENV/bin:$PATH"
+ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6"
+ENV TCNN_CUDA_ARCHITECTURES=86;80;75;70;61;60
+ENV FORCE_CUDA=1
 
-# Set environment variables
+ENV CUDA_HOME=/usr/local/cuda
+ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:${PATH}
+ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
+ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH}
+
+# Set home to the user's home directory
 ENV HOME=/home/user \
-    CUDA_HOME=/usr/local/cuda \
-    LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} \
-    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
+	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 working directory
 WORKDIR $HOME/app