Spaces:
Running
Running
Include original image in gallery output
Browse files
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 |
|