Update app.py
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ def detect_document(image):
|
|
62 |
x1, y1, x2, y2 = map(int, box.xyxy[0])
|
63 |
conf = box.conf[0]
|
64 |
cls = int(box.cls[0])
|
65 |
-
class_name =
|
66 |
|
67 |
detected_classes.add(class_name)
|
68 |
label = f"{class_name} {conf:.2f}"
|
|
|
62 |
x1, y1, x2, y2 = map(int, box.xyxy[0])
|
63 |
conf = box.conf[0]
|
64 |
cls = int(box.cls[0])
|
65 |
+
class_name = modelY.names[cls]
|
66 |
|
67 |
detected_classes.add(class_name)
|
68 |
label = f"{class_name} {conf:.2f}"
|