fix dep req.txt optimize image size

.dockerignore

@@ -1,8 +1,19 @@
 __pycache__
 *.pyc
-.git
-.env
-venv
+*.pyo
+*.pyd
+.Python
+env/
+venv/
 .venv
-node_modules
+.env
+.git
+.gitignore
+.vscode
+.idea
+docker_data/
+output/
+*.pdf
+*.jpg
+*.png
 .cache

Dockerfile

@@ -1,6 +1,6 @@
 FROM python:3.11-slim
 
-WORKDIR /home/user/app
+WORKDIR /home/user
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
     wget gnupg ca-certificates \
@@ -16,14 +16,15 @@ RUN pip install --no-cache-dir playwright && \
     PLAYWRIGHT_BROWSERS_PATH=/home/user/.cache/ms-playwright playwright install chromium --with-deps && \
     rm -rf /var/lib/apt/lists/*
 
-COPY . .
-
-RUN chmod -R 755 /home/user
-
-EXPOSE 7860
 ENV PORT=7860
 ENV PLAYWRIGHT_BROWSERS_PATH=/home/user/.cache/ms-playwright
-
 ENV HF_HOME=/home/user/.cache/huggingface
+RUN chmod -R 755 /home/user
+
+RUN python -c "from transformers import AutoModelForImageSegmentation; AutoModelForImageSegmentation.from_pretrained('briaai/RMBG-1.4', trust_remote_code=True)"
+
+COPY . .
+
+EXPOSE 7860
 
 CMD ["python", "-m", "hypercorn", "server:app", "-b", "0.0.0.0:7860", "--access-log", "-", "--error-log", "-", "--log-level", "info"]

requirements.txt

@@ -1,16 +1,12 @@
 python-dotenv
 requests
 playwright
-asyncio
-dotenv
-Flask
 playwright-stealth
 hypercorn
 google-genai
 beautifulsoup4
 pymupdf
 flask[async]
-pandas
 fpdf
 pandas
 
@@ -23,5 +19,4 @@ numpy
 opencv-python-headless
 pillow
 transformers
-huggingface_hub
 accelerate