Sergey Mikhno
commited on
Commit
·
8986a3c
1
Parent(s):
b90b21f
commit
Browse files- app/app.py +1 -1
app/app.py
CHANGED
|
@@ -10,7 +10,7 @@ from werkzeug.utils import secure_filename
|
|
| 10 |
UPLOAD_FOLDER = 'upload'
|
| 11 |
ALLOWED_EXTENSIONS = {'txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'}
|
| 12 |
|
| 13 |
-
filepath = hf_hub_download(repo_id="juliozhao/DocLayout-YOLO-DocStructBench", filename="doclayout_yolo_docstructbench_imgsz1024.pt")
|
| 14 |
model = YOLOv10(filepath)
|
| 15 |
|
| 16 |
app = Flask(__name__)
|
|
|
|
| 10 |
UPLOAD_FOLDER = 'upload'
|
| 11 |
ALLOWED_EXTENSIONS = {'txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'}
|
| 12 |
|
| 13 |
+
filepath = hf_hub_download(repo_id="juliozhao/DocLayout-YOLO-DocStructBench", filename="doclayout_yolo_docstructbench_imgsz1024.pt", local_dir="/tmp")
|
| 14 |
model = YOLOv10(filepath)
|
| 15 |
|
| 16 |
app = Flask(__name__)
|