905saini commited on
Commit
9d71789
·
1 Parent(s): 270b39e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -35,6 +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
  if class_name =="Header":
39
  color = (0,0,255) #Blue color
40
  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.subheader(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)