syedfaisalabrar commited on
Commit
c68f3a4
·
verified ·
1 Parent(s): af02505

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = model.names[cls]
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}"