Update Dockerfile

Dockerfile

@@ -1,18 +1,22 @@
-# Set up base image (use the correct Hugging Face registry)
-FROM registry.huggingface.co\space\microsoft-omniparser:latest
+FROM nvidia/cuda:12.2.2-runtime-ubuntu22.04
 
+ENV DEBIAN_FRONTEND=noninteractive
 
-USER root
+RUN apt-get update -q && \
+    apt-get install -y --no-install-recommends \
+    python3-pip \
+    python3-dev \
+    ffmpeg \
+    libgl1 \
+    git \
+    wget \
+    && rm -rf /var/lib/apt/lists/*
 
-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
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
 
-RUN pip install fastapi[all] 
+COPY . .
 
+EXPOSE 7860
 
-COPY main.py main.py
-RUN python main.py
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]