Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ def process_img(image):
|
|
27 |
with torch.no_grad():
|
28 |
result = model(source=image)
|
29 |
if len(result[0].boxes)>0:
|
30 |
-
ann=Annotator(im=
|
31 |
boxes=result[0].boxes.xyxy
|
32 |
for element in boxes:
|
33 |
box=np.array(element.cpu()).flatten()
|
@@ -48,7 +48,7 @@ def process_img(image):
|
|
48 |
ann.box_label(box=box, label=label, color=(0,128,0))
|
49 |
vis=ann.result()
|
50 |
else:
|
51 |
-
vis =
|
52 |
return vis
|
53 |
|
54 |
title = "Efficient Hazy Vehicle Detection ✏️[] 🤗"
|
|
|
27 |
with torch.no_grad():
|
28 |
result = model(source=image)
|
29 |
if len(result[0].boxes)>0:
|
30 |
+
ann=Annotator(im=image)
|
31 |
boxes=result[0].boxes.xyxy
|
32 |
for element in boxes:
|
33 |
box=np.array(element.cpu()).flatten()
|
|
|
48 |
ann.box_label(box=box, label=label, color=(0,128,0))
|
49 |
vis=ann.result()
|
50 |
else:
|
51 |
+
vis = image
|
52 |
return vis
|
53 |
|
54 |
title = "Efficient Hazy Vehicle Detection ✏️[] 🤗"
|