Update app.py
Browse files
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("
|
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)
|