Update Dockerfile

Dockerfile

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