Spaces:

guiwitz
/

yolo5demo

Runtime error

guiwitz commited on Oct 10, 2022

Commit

c48f318

1 Parent(s): 5a21fa9

fix model input

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ from PIL import Image
 model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
 def inference(im):
-    results = model(imgs)
     results.render()  # updates results.imgs with boxes and labels

 model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
 def inference(im):
+    results = model(im)
     results.render()  # updates results.imgs with boxes and labels