Spaces:
Runtime error
Runtime error
adding the doclaynet finetuned model
Browse files
app.py
CHANGED
@@ -60,7 +60,8 @@ if __name__ == "__main__":
|
|
60 |
# == load model ==
|
61 |
from doclayout_yolo import YOLOv10
|
62 |
print(f"Using device: {device}")
|
63 |
-
model = YOLOv10(os.path.join(os.path.dirname(__file__), "models", "DocLayout-YOLO-DocStructBench", "doclayout_yolo_docstructbench_imgsz1024.pt")) # load an official model
|
|
|
64 |
|
65 |
with open("header.html", "r") as file:
|
66 |
header = file.read()
|
|
|
60 |
# == load model ==
|
61 |
from doclayout_yolo import YOLOv10
|
62 |
print(f"Using device: {device}")
|
63 |
+
# model = YOLOv10(os.path.join(os.path.dirname(__file__), "models", "DocLayout-YOLO-DocStructBench", "doclayout_yolo_docstructbench_imgsz1024.pt")) # load an official model
|
64 |
+
model = YOLOv10(os.path.join(os.path.dirname(__file__), "models", "DocLayout-YOLO-DocLayNet-Docsynth300K_pretrained", "doclayout_yolo_doclaynet_imgsz1120_docsynth_pretrain.pt")) # load an official model
|
65 |
|
66 |
with open("header.html", "r") as file:
|
67 |
header = file.read()
|