commit

Dockerfile

@@ -7,4 +7,4 @@ COPY ./app .
 RUN pip3 install -r requirements.txt
 ENV MPLCONFIGDIR=/tmp/MPLCONFIGDIR/
 ENV YOLO_CONFIG_DIR=/tmp/YOLO
-CMD gunicorn --bind 0.0.0.0:8888 app:app
+CMD ["gunicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app/app.py

@@ -10,7 +10,7 @@ from werkzeug.utils import secure_filename
 UPLOAD_FOLDER = 'upload'
 ALLOWED_EXTENSIONS = {'txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'}
 
-filepath = hf_hub_download(repo_id="juliozhao/DocLayout-YOLO-DocStructBench", filename="doclayout_yolo_docstructbench_imgsz1024.pt", local_dir="/tmp")
+filepath = hf_hub_download(repo_id="juliozhao/DocLayout-YOLO-DocStructBench", filename="doclayout_yolo_docstructbench_imgsz1024.pt", local_dir="/tmp", cache_dir="/tmp")
 model = YOLOv10(filepath)
 
 app = Flask(__name__)