Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ def preprocessing(image):
|
|
24 |
# YOLO document detection and cropping
|
25 |
def detect_and_crop_document(image):
|
26 |
image_np = np.array(image)
|
27 |
-
results = modelY(image_np, conf=0.
|
28 |
cropped_images = []
|
29 |
predictions = []
|
30 |
|
|
|
24 |
# YOLO document detection and cropping
|
25 |
def detect_and_crop_document(image):
|
26 |
image_np = np.array(image)
|
27 |
+
results = modelY(image_np, conf=0.80, device='cpu')
|
28 |
cropped_images = []
|
29 |
predictions = []
|
30 |
|