Spaces:

guru001
/

yolov3-coco-torch

Sleeping

guru001 commited on Sep 28, 2023

Commit

c9644ce

•

1 Parent(s): 4968292

PIL removed

Files changed (1) hide show

app.py CHANGED Viewed

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

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