Spaces:

guiwitz
/

yolo5demo

Runtime error

guiwitz commited on Oct 10, 2022

Commit

86f7280

1 Parent(s): c48f318

fix ims

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ def inference(im):
     results.render()  # updates results.imgs with boxes and labels
-    return Image.fromarray(results.imgs[0])
 inputs = gr.inputs.Image(type='pil', label="Original Image")

     results.render()  # updates results.imgs with boxes and labels
+    return Image.fromarray(results.ims[0])
 inputs = gr.inputs.Image(type='pil', label="Original Image")