Update Dockerfile

Dockerfile

@@ -2,7 +2,10 @@ FROM registry.hf.space/microsoft-omniparser:latest
 
 USER root
 
-# Install CUDA dependencies
+# Set proper permissions for /tmp
+RUN chmod 1777 /tmp
+
+# Install CUDA dependencies and system packages
 RUN apt update -q && apt install -y --no-install-recommends \
     ca-certificates wget libgl1 \
     && wget -qO /tmp/cuda-keyring.deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb \
@@ -21,5 +24,5 @@ COPY weights.py weights.py
 # Expose application port
 EXPOSE 7860
 
-# Set the entrypoint for the container
+# Run weights.py and start the FastAPI server
 CMD ["bash", "-c", "python weights.py && uvicorn main:app --host 0.0.0.0 --port 7860"]