Update Dockerfile

Dockerfile

@@ -1,29 +1,17 @@
-# Use official Python image
-FROM python:3.9-slim
+FROM registry.hf.space/microsoft-omniparser:latest
 
-# Install system dependencies
-RUN apt update -q && apt install -y \
-    ca-certificates \
-    wget \
-    libgl1 \
-    && rm -rf /var/lib/apt/lists/*
 
-# Install Python dependencies
-RUN pip install --no-cache-dir \
-    fastapi[all] \
-    torch \
-    ultralytics \
-    transformers \
-    pillow \
-    python-multipart \
-    opencv-python-headless \
-    easyocr \
-    paddleocr
+USER root
 
-# Copy application code
-COPY main.py /app/main.py
-COPY utils.py /app/utils.py
-WORKDIR /app
+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
 
-# Expose port and run application
+RUN pip install fastapi[all] 
+
+
+COPY main.py main.py
+RUN python main.py
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]