Spaces:

heerjtdev
/

yolo_layoutlm

Sleeping

heerjtdev commited on Nov 18

Commit

75059b9

1 Parent(s): 01bdba2

Update working_yolo_pipeline.py

Files changed (1) hide show

working_yolo_pipeline.py CHANGED Viewed

@@ -27,7 +27,7 @@ import tempfile  # Recommended for robust temporary file handling
 # NOTE: Update these paths to match your environment before running!
 WEIGHTS_PATH = 'YOLO_MATH/yolo_split_data/runs/detect/math_figure_detector_v3/weights/best.pt'
-DEFAULT_LAYOUTLMV3_MODEL_PATH = "checkpoints/layoutlmv3_trained_20251118.pth"
 # DIRECTORY CONFIGURATION
 # NOTE: These are now used for temporary data extraction/storage

 # NOTE: Update these paths to match your environment before running!
 WEIGHTS_PATH = 'YOLO_MATH/yolo_split_data/runs/detect/math_figure_detector_v3/weights/best.pt'
+DEFAULT_LAYOUTLMV3_MODEL_PATH = "layoutlmv3_trained_20251118.pth"
 # DIRECTORY CONFIGURATION
 # NOTE: These are now used for temporary data extraction/storage