Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|