Spaces:

ananthvk
/

fruits-classifier

Sleeping

ananthvk commited on Mar 9

Commit

4e51fae

1 Parent(s): a4e002d

commit

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ def classify_fruit(img):
     pred, idx, probs = learn.predict(img)
     return dict(zip(categories, map(float, probs)))
-image = gr.Image(width=224, height=224)
 label = gr.Label()
 inter = gr.Interface(fn=classify_fruit, inputs=image, outputs=label)

     pred, idx, probs = learn.predict(img)
     return dict(zip(categories, map(float, probs)))
+image = gr.Image(width=224, height=224, type="pil")
 label = gr.Label()
 inter = gr.Interface(fn=classify_fruit, inputs=image, outputs=label)