runtime model download

Dockerfile

@@ -6,13 +6,13 @@ COPY smishing_detector/requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt && pip install gunicorn huggingface_hub
 
 COPY smishing_detector/ ./smishing_detector/
-RUN python -c "from huggingface_hub import hf_hub_download; \
-    hf_hub_download(repo_id='ExistedYear/scamshield_ai', filename='best_model.pt', \
-    local_dir='/app/smishing_detector', local_dir_use_symlinks=False)"
 
 WORKDIR /app/smishing_detector
 ENV SCAMSHIELD_DEBUG=false
 ENV SCAMSHIELD_HOST=0.0.0.0
 ENV SCAMSHIELD_PORT=7860
 
-CMD ["gunicorn", "-w", "1", "-b", "0.0.0.0:7860", "--timeout", "120", "app.flask_api:app"]
+COPY entrypoint.sh /app/entrypoint.sh
+RUN chmod +x /app/entrypoint.sh
+
+CMD ["/app/entrypoint.sh"]

entrypoint.sh

@@ -0,0 +1,8 @@
+#!/bin/bash
+set -e
+
+python -c "from huggingface_hub import hf_hub_download; \
+    hf_hub_download(repo_id='ExistedYear/scamshield_ai', filename='best_model.pt', \
+    local_dir='/app/smishing_detector', local_dir_use_symlinks=False)"
+
+exec gunicorn -w 1 -b 0.0.0.0:7860 --timeout 120 app.flask_api:app