Spaces:

KamrusSamad
/

object-recognizer-for-vehicle

Runtime error

KamrusSamad commited on Jan 24, 2023

Commit

e80e252

1 Parent(s): 5adcee7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,7 +23,7 @@ model = load_learner('obj-recognizer-v2.pkl')
 def recognize_image(image):
     pred, idx, probs = model.predict(image)
-    return dict(zip(obj_labels, map(float, probs)))
 image = gr.inputs.Image(shape=(192,192))
 label = gr.outputs.Label(num_top_classes=5)

 def recognize_image(image):
     pred, idx, probs = model.predict(image)
+    return pred
 image = gr.inputs.Image(shape=(192,192))
 label = gr.outputs.Label(num_top_classes=5)