Hamza-cpp commited on
Commit
4056dea
·
verified ·
1 Parent(s): e2544b2

adding the doclaynet finetuned model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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()