Edvin Behdadijd
commited on
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)
|