cjlise commited on
Commit
a76d56e
·
1 Parent(s): 7721e0f

Fix classify_image error

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -19,11 +19,11 @@ learn = load_learner('model.pkl')
19
  # finally:
20
  # pathlib.PosixPath = posix_backup
21
 
22
- # # %% ..\lesson2-app.ipynb 6
23
- # categories = ('Dog','Cat')
24
- # def classify_image(img):
25
- # pred, idx, probs = learn.predict(img)
26
- # return dict(zip(categories, map(float,probs)))
27
 
28
  # %% ..\lesson2-app.ipynb 8
29
  image = gr.inputs.Image(shape=(192,192))
 
19
  # finally:
20
  # pathlib.PosixPath = posix_backup
21
 
22
+ # %% ..\lesson2-app.ipynb 6
23
+ categories = ('Dog','Cat')
24
+ def classify_image(img):
25
+ pred, idx, probs = learn.predict(img)
26
+ return dict(zip(categories, map(float, probs)))
27
 
28
  # %% ..\lesson2-app.ipynb 8
29
  image = gr.inputs.Image(shape=(192,192))