Spaces:

offthewallace
/

libYOLODoc

Runtime error

offthewallace commited on Sep 12, 2022

Commit

70063f9

1 Parent(s): 3ff55ee

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ def yolo(im, size=640):
     results = model(im)  # inference
     results.render()  # updates results.imgs with boxes and labels
-    return Image.fromarray(results.imgs[0])
 def detect(image):
     results = model(image)

     results = model(im)  # inference
     results.render()  # updates results.imgs with boxes and labels
+    return Image.fromarray(results.ims[0])
 def detect(image):
     results = model(image)