Spaces:

guru001
/

yolov3-coco-torch

Sleeping

guru001 commited on Sep 28, 2023

Commit

4968292

•

1 Parent(s): 302e878

input_img added

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ model = yolo_model()
 def predict(input_img):
     input_img = Image.fromarray(input_img)
-    _, payload = model.predict(image)
     # print('prediction',prediction)
     return payload

 def predict(input_img):
     input_img = Image.fromarray(input_img)
+    _, payload = model.predict(input_img)
     # print('prediction',prediction)
     return payload