Spaces:
Sleeping
Sleeping
Commit
·
3de965c
1
Parent(s):
1fa413d
small change
Browse files- working_yolo_pipeline.py +2 -2
working_yolo_pipeline.py
CHANGED
|
@@ -26,8 +26,8 @@ import tempfile # Recommended for robust temporary file handling
|
|
| 26 |
# ============================================================================
|
| 27 |
|
| 28 |
# NOTE: Update these paths to match your environment before running!
|
| 29 |
-
WEIGHTS_PATH = '
|
| 30 |
-
DEFAULT_LAYOUTLMV3_MODEL_PATH = "checkpoints/
|
| 31 |
|
| 32 |
# DIRECTORY CONFIGURATION
|
| 33 |
# NOTE: These are now used for temporary data extraction/storage
|
|
|
|
| 26 |
# ============================================================================
|
| 27 |
|
| 28 |
# NOTE: Update these paths to match your environment before running!
|
| 29 |
+
WEIGHTS_PATH = 'YOLO_MATH/yolo_split_data/runs/detect/math_figure_detector_v3/weights/best.pt'
|
| 30 |
+
DEFAULT_LAYOUTLMV3_MODEL_PATH = "checkpoints/layoutlmv3_trained_20251110_111123.pth"
|
| 31 |
|
| 32 |
# DIRECTORY CONFIGURATION
|
| 33 |
# NOTE: These are now used for temporary data extraction/storage
|