Spaces:

offthewallace
/

libYOLODoc

Runtime error

offthewallace commited on Sep 12, 2022

Commit

3c3369a

1 Parent(s): 70063f9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ def yolo(im, size=640):
 def detect(image):
     results = model(image)
     results.render()
-    return Image.fromarray(results.imgs[0])
 inputs = gr.inputs.Image(type='pil', label="Original Image")

 def detect(image):
     results = model(image)
     results.render()
+    return Image.fromarray(results.ims[0])
 inputs = gr.inputs.Image(type='pil', label="Original Image")