Update Dockerfile

Dockerfile

@@ -17,6 +17,12 @@ USER user
 # Set home to the user's home directory
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
+
+
+# Define environment variables for GPU, RAM limit, and shared memory
+ENV GPU_ID="0" \
+    RAM_LIMIT="4g" \
+    SHM_SIZE="2g"
  
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app