Spaces:
Runtime error
Runtime error
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 = []
|