Update Dockerfile

Dockerfile

@@ -2,27 +2,15 @@ FROM registry.hf.space/microsoft-omniparser:latest
 
 USER root
 
-# 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 \
+RUN chmod 1777 /tmp \
+    && apt update -q && apt install -y 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 \
     && dpkg -i /tmp/cuda-keyring.deb && apt update -q \
-    && apt install -y --no-install-recommends libcudnn8 libcublas-12-2 \
-    && rm -rf /var/lib/apt/lists/* /tmp/cuda-keyring.deb
-
-# Install Python dependencies
-COPY requirements.txt requirements.txt
-RUN pip install --no-cache-dir -r requirements.txt
+    && apt install -y --no-install-recommends libcudnn8 libcublas-12-2
 
-# Copy application files
-COPY main.py main.py
-COPY weights.py weights.py
+RUN pip install fastapi[all] 
 
-# Expose application port
-EXPOSE 7860
 
-# Run weights.py and start the FastAPI server
-CMD ["bash", "-c", "python weights.py && uvicorn main:app --host 0.0.0.0 --port 7860"]
+COPY main.py main.py
+RUN python main.py
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]