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 |
|