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