Hugging Face's logo

Spaces:
heerjtdev
/
yolo_layoutlm
Build error

App Files Community
heerjtdev commited on
Commit
950cc9f
·
1 Parent(s): 35de6cd

Update working_yolo_pipeline.py

Browse files
Files changed (1) hide show
  1. working_yolo_pipeline.py +1 -1
working_yolo_pipeline.py CHANGED
@@ -2916,7 +2916,7 @@ import shutil
2916
 
2917
  # NOTE: Update these paths to match your environment before running!
2918
  WEIGHTS_PATH = 'YOLO_MATH/yolo_split_data/runs/detect/math_figure_detector_v3/weights/best.pt'
2919
- DEFAULT_LAYOUTLMV3_MODEL_PATH = "layoutlmv3_trained_20251118.pth"
2920
 
2921
  # DIRECTORY CONFIGURATION
2922
  OCR_JSON_OUTPUT_DIR = './ocr_json_output_final'
 
2916
 
2917
  # NOTE: Update these paths to match your environment before running!
2918
  WEIGHTS_PATH = 'YOLO_MATH/yolo_split_data/runs/detect/math_figure_detector_v3/weights/best.pt'
2919
+ DEFAULT_LAYOUTLMV3_MODEL_PATH = "97.pth"
2920
 
2921
  # DIRECTORY CONFIGURATION
2922
  OCR_JSON_OUTPUT_DIR = './ocr_json_output_final'