Update Dockerfile

Dockerfile

@@ -1,21 +1,22 @@
 FROM python:3.10
 
-# Set the working directory
-WORKDIR /app
+# 1. Create a writable cache directory for huggingface_hub
+RUN mkdir -p /tmp/huggingface_cache
+ENV HUGGINGFACE_HUB_CACHE=/tmp/huggingface_cache
 
-# Set writable cache for Numba to prevent caching errors
-ENV NUMBA_CACHE_DIR=/tmp/numba_cache/
+# 2. Create a writable cache directory for Numba (used by rembg/pymatting)
+RUN mkdir -p /tmp/numba_cache
+ENV NUMBA_CACHE_DIR=/tmp/numba_cache
 
-# Install dependencies
+WORKDIR /app
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy application files
 COPY . /app/
 
-# Expose the port FastAPI will run on
-EXPOSE 8000
+# 3. Expose port 7860 for Hugging Face
+EXPOSE 7860
 
-# Run FastAPI when the container starts
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
+# 4. Run your main.py (which calls uvicorn.run inside the __main__ block)
+CMD ["python", "main.py"]