cdcvd commited on
Commit
91717a7
·
verified ·
1 Parent(s): 1ca70a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -75,10 +75,10 @@ def process_file(input_file):
75
  def run_detection_and_ocr():
76
  # Load models
77
  ocr_model = Model.load('hezarai/crnn-fa-printed-96-long')
78
- yolo_model = YOLO("best.pt")
79
 
80
  input_folder = 'output_images'
81
- yolo_model.predict(input_folder, save=True, imgsz=320, conf=0.5, save_crop=True)
82
 
83
  output_folder = 'runs/detect/predict' # Remove leading slash if needed
84
  crop_folder = os.path.join(output_folder, 'crops')
 
75
  def run_detection_and_ocr():
76
  # Load models
77
  ocr_model = Model.load('hezarai/crnn-fa-printed-96-long')
78
+ yolo_model = YOLO("best_300_D_check.pt")
79
 
80
  input_folder = 'output_images'
81
+ yolo_model.predict(input_folder, save=True, conf=0.5, save_crop=True)
82
 
83
  output_folder = 'runs/detect/predict' # Remove leading slash if needed
84
  crop_folder = os.path.join(output_folder, 'crops')