Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ import numpy as np
|
|
7 |
# Define the detect function
|
8 |
def detect_and_crop(input_image):
|
9 |
# Define paths and parameters
|
10 |
-
weights_path = 'yolo/yolov7-main/
|
11 |
img_size = 640
|
12 |
conf = 0.20
|
13 |
source = 'dataset/images/train/' # Folder for input images
|
|
|
7 |
# Define the detect function
|
8 |
def detect_and_crop(input_image):
|
9 |
# Define paths and parameters
|
10 |
+
weights_path = 'yolo/yolov7-main/best.pt'
|
11 |
img_size = 640
|
12 |
conf = 0.20
|
13 |
source = 'dataset/images/train/' # Folder for input images
|