jph00 commited on
Commit
47197cc
1 Parent(s): 05ce80a
Files changed (3) hide show
  1. .gitignore +4 -0
  2. app.ipynb +0 -0
  3. app.py +11 -3
.gitignore ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ .ipynb_checkpoints/
2
+ *.bak
3
+ *.swp
4
+
app.ipynb ADDED
The diff for this file is too large to render. See raw diff
 
app.py CHANGED
@@ -1,19 +1,27 @@
 
 
 
 
 
1
  from fastai.vision.all import *
2
  import gradio as gr
3
 
4
  def is_cat(x): return x[0].isupper()
5
 
 
6
  learn = load_learner('model.pkl')
 
 
7
  categories = ('Dog', 'Cat')
8
 
9
  def classify_image(img):
10
  pred,idx,probs = learn.predict(img)
11
  return dict(zip(categories, map(float,probs)))
12
 
13
- image = gr.inputs.Image(shape=(192, 192))
 
14
  label = gr.outputs.Label()
15
  examples = ['dog.jpg', 'cat.jpg', 'dunno.jpg']
16
 
17
  intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
18
- intf.launch()
19
-
 
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=(128, 128))
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()