Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -68,7 +68,7 @@ def imageRotation(image):
|
|
68 |
def detect_document(image):
|
69 |
"""Detect front/back of the document using YOLO."""
|
70 |
image_np = np.array(image)
|
71 |
-
results = modelY(image_np, conf=0.
|
72 |
|
73 |
detected_classes = set()
|
74 |
labels = []
|
|
|
68 |
def detect_document(image):
|
69 |
"""Detect front/back of the document using YOLO."""
|
70 |
image_np = np.array(image)
|
71 |
+
results = modelY(image_np, conf=0.70, device='cpu')
|
72 |
|
73 |
detected_classes = set()
|
74 |
labels = []
|