Update Dockerfile

Dockerfile

@@ -1,10 +1,14 @@
+# Set up base image (use the correct Hugging Face registry)
+FROM registry.huggingface.co/microsoft/omniparser:latest
+
+# Set environment variables for authentication (persist across layers)
 ARG HF_TOKEN
 ARG HF_USERNAME
+ENV HF_TOKEN=${HF_TOKEN}
+ENV HF_USERNAME=${HF_USERNAME}
 
-RUN echo "$HF_TOKEN" | docker login registry.huggingface.co -u $HF_USERNAME --password-stdin
-
-# Use the correct Hugging Face container
-FROM registry.hf.space/microsoft-omniparser:latest
+# Authenticate with Hugging Face's Docker registry
+RUN echo "$HF_TOKEN" | docker login registry.huggingface.co -u "$HF_USERNAME" --password-stdin
 
 # Switch to root user for package installation
 USER root
@@ -17,11 +21,14 @@ RUN chmod 1777 /tmp \
     && apt install -y --no-install-recommends libcudnn8 libcublas-12-2
 
 # Install required Python packages
-RUN pip install fastapi[all] 
+RUN pip install fastapi[all]
 
-# Ensure main.py exists in the same directory as the Dockerfile
-COPY main.py /app/main.py
+# Create the application directory
+RUN mkdir -p /app
 WORKDIR /app
 
-# Run the script
+# Copy application files
+COPY main.py /app/main.py
+
+# Set entry point to run the FastAPI server
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]