heerjtdev commited on
Commit
898e575
·
1 Parent(s): 75c8d92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,8 +30,8 @@ from sklearn.metrics.pairwise import cosine_similarity
30
  # ============================================================================
31
 
32
  # NOTE: Update these paths to match your environment before running!
33
- WEIGHTS_PATH = 'YOLO_MATH/yolo_split_data/runs/detect/math_figure_detector_v3/weights/best.pt'
34
- DEFAULT_LAYOUTLMV3_MODEL_PATH = "97.pth"
35
 
36
  # DIRECTORY CONFIGURATION
37
  OCR_JSON_OUTPUT_DIR = './ocr_json_output_final'
 
30
  # ============================================================================
31
 
32
  # NOTE: Update these paths to match your environment before running!
33
+ WEIGHTS_PATH = 'best.pt'
34
+ DEFAULT_LAYOUTLMV3_MODEL_PATH = "98.pth"
35
 
36
  # DIRECTORY CONFIGURATION
37
  OCR_JSON_OUTPUT_DIR = './ocr_json_output_final'