Spaces:

0xnbk
/

is-cat

Sleeping

0xnbk commited on Aug 7

Commit

8c64360

1 Parent(s): cb56147

Fix for build error

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ learn = load_learner('model.pkl')
 categories = ('Dog', 'Cat')
 def classify_image(img):
-    pred,idx,probs = learn.predict(im)
     return dict(zip(categories, map(float, probs)))
 # %% ../app.ipynb 8

 categories = ('Dog', 'Cat')
 def classify_image(img):
+    pred,idx,probs = learn.predict(img)
     return dict(zip(categories, map(float, probs)))
 # %% ../app.ipynb 8