Spaces:

skcoirz
/

fastai2

Runtime error

skcoirz commited on Apr 14, 2023

Commit

1156a91

1 Parent(s): 51b5d35

fix shape typo

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ def infer(img):
     _, _, prob = learner.predict(img)
     return dict(zip(categories, map(float, prob)))
-image = gr.inputs.Image(shape(192, 192))
 label = gr.outputs.Label()
 iface = gr.Interface(fn=infer, inputs=image, outputs=label)

     _, _, prob = learner.predict(img)
     return dict(zip(categories, map(float, prob)))
+image = gr.inputs.Image(shape = (192, 192))
 label = gr.outputs.Label()
 iface = gr.Interface(fn=infer, inputs=image, outputs=label)