etiennefd commited on
Commit
54b5e2e
1 Parent(s): 29fffb9
Files changed (2) hide show
  1. .DS_Store +0 -0
  2. app.py +20 -4
.DS_Store ADDED
Binary file (6.15 kB). View file
 
app.py CHANGED
@@ -1,7 +1,23 @@
 
 
 
1
  import gradio as gr
2
 
3
- def greet(name):
4
- return "Hello " + name + "!!"
 
 
 
 
 
 
 
 
 
 
 
 
 
5
 
6
- iface = gr.Interface(fn=greet, inputs="text", outputs="text")
7
- iface.launch()
 
1
+ #| export
2
+
3
+ from fastai.vision.all import *
4
  import gradio as gr
5
 
6
+ def is_cat(x): return x[0].isupper() # need this redefinition because the learner uses this function
7
+
8
+ learn = load_learner('model.pkl')
9
+
10
+ categories = ('Dog', 'Cat')
11
+
12
+ # turn the output into a dict with categories and probabilities
13
+ def classify_image(img):
14
+ pred, index, probs = learn.predict(img)
15
+ return dict(zip(categories, map(float, probs)))
16
+
17
+ # gradio interface
18
+ image = gr.inputs.Image(shape=(192, 192))
19
+ label = gr.outputs.Label()
20
+ examples = ['dog.jpg', 'cat.jpg', 'dunno.jpg']
21
 
22
+ interface = gr.Interface(fn = classify_image, inputs = image, outputs = label, examples = examples)
23
+ interface.launch(inline = False)