wmacdonald commited on
Commit
dda4914
1 Parent(s): e4dd32b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -27
app.py CHANGED
@@ -1,27 +0,0 @@
1
- # AUTOGENERATED! DO NOT EDIT! File to edit: . (unless otherwise specified).
2
-
3
- __all__ = ['is_cat', 'learn', 'classify_image', 'categories', 'image', 'label', 'examples', 'intf']
4
-
5
- # Cell
6
- from fastai.vision.all import *
7
- import gradio as gr
8
-
9
- def is_cat(x): return x[0].isupper()
10
-
11
- # Cell
12
- learn = load_learner('model.pkl')
13
-
14
- # Cell
15
- categories = ('Dog', 'Cat')
16
-
17
- def classify_image(img):
18
- pred,idx,probs = learn.predict(img)
19
- return dict(zip(categories, map(float,probs)))
20
-
21
- # Cell
22
- image = gr.inputs.Image(shape=(192, 192))
23
- label = gr.outputs.Label()
24
- examples = ['dog.jpg', 'cat.jpg', 'dunno.jpg']
25
-
26
- intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
27
- intf.launch(inline=False)