905saini commited on
Commit
903cf0c
·
1 Parent(s): 60b6939

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ if file_name is not None:
35
  if confidance[i] >= threshold:
36
  x1,y1,x2,y2 = bbox[i]*640
37
  class_name = labels[int(classes[i])]
38
- st.title(class_name)
39
  if class_name =="Header":
40
  color = (0,0,255) #Blue color
41
  cv2.rectangle(copy_img, (int(x1), int(y1)), (int(x2), int(y2)),color, 2)
 
35
  if confidance[i] >= threshold:
36
  x1,y1,x2,y2 = bbox[i]*640
37
  class_name = labels[int(classes[i])]
38
+ st.text(class_name+" : "+confidance[i]+"%")
39
  if class_name =="Header":
40
  color = (0,0,255) #Blue color
41
  cv2.rectangle(copy_img, (int(x1), int(y1)), (int(x2), int(y2)),color, 2)