rdezwart commited on
Commit
33f25f1
·
1 Parent(s): 1bb134f

Include original image in gallery output

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -69,6 +69,7 @@ def detect_objects(img: Image.Image):
69
  f"{yolos_model.config.id2label[label.item()]} ({round(score.item(), 3)})")
70
  )
71
 
 
72
  return box_images
73
 
74
 
 
69
  f"{yolos_model.config.id2label[label.item()]} ({round(score.item(), 3)})")
70
  )
71
 
72
+ box_images.append((img, f"original"))
73
  return box_images
74
 
75