Update app.py
Browse files
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("
|
79 |
|
80 |
input_folder = 'output_images'
|
81 |
-
yolo_model.predict(input_folder, save=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')
|