paugar commited on
Commit
efc152c
·
verified ·
1 Parent(s): 2ea737d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=img)
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 = img
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 ✏️[] 🤗"