Spaces:

toobarah
/

minimal

Build error

toobarah commited on May 2

Commit

95fd352

•

1 Parent(s): 059596b

fastai library added

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ categories = ('dog', 'cat')
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
-    return dict(zip(categories, map(float, probs)))
 image = gr.inputs.Image(shape=(256,256))
 label = gr.outputs.Label()

 def classify_image(img):
     pred,idx,probs = learn.predict(img)
+    return dict(zip(categories, map(float, probs)))
 image = gr.inputs.Image(shape=(256,256))
 label = gr.outputs.Label()