cdcvd commited on
Commit
a12c877
·
verified ·
1 Parent(s): f99f042

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -76,7 +76,7 @@ def process_file(input_file):
76
  def run_detection_and_ocr():
77
  # Load models
78
  ocr_model = Model.load('hezarai/crnn-fa-printed-96-long')
79
- yolo_model = YOLO("/best.pt")
80
 
81
  input_folder = 'output_images'
82
  yolo_model.predict(input_folder, save=True, imgsz=320, conf=0.5, save_crop=True)
 
76
  def run_detection_and_ocr():
77
  # Load models
78
  ocr_model = Model.load('hezarai/crnn-fa-printed-96-long')
79
+ yolo_model = YOLO("best.pt")
80
 
81
  input_folder = 'output_images'
82
  yolo_model.predict(input_folder, save=True, imgsz=320, conf=0.5, save_crop=True)