Update Dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-FROM registry.huggingface.co/spaces/microsoft/omniparser:latest
+FROM python:3.10  # Use Python instead of missing omniparser image
 
 USER root
 
@@ -8,9 +8,14 @@ RUN chmod 1777 /tmp \
     && dpkg -i /tmp/cuda-keyring.deb && apt update -q \
     && apt install -y --no-install-recommends libcudnn8 libcublas-12-2
 
-RUN pip install fastapi[all] 
-
+# Install FastAPI and other Python dependencies
+RUN pip install fastapi[all] uvicorn
 
+# Copy application code
 COPY main.py main.py
-RUN python main.py
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+
+# Run Python script to check for errors before starting FastAPI
+RUN python main.py  
+
+# Start FastAPI server
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]