Abyszero commited on
Commit
7ec28e7
1 Parent(s): 7c65005

Switch to cat vs dog example

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. model.pkl +2 -2
app.py CHANGED
@@ -1,7 +1,8 @@
1
  import gradio as gr
2
  from fastai.vision.all import *
3
 
4
- categories = ('ninja', 'not ninja')
 
5
  learn = load_learner('model.pkl')
6
 
7
  def classify_image(img):
 
1
  import gradio as gr
2
  from fastai.vision.all import *
3
 
4
+ def is_cat(x): return x[0].isupper()
5
+ categories = ('cat', 'dog')
6
  learn = load_learner('model.pkl')
7
 
8
  def classify_image(img):
model.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f4972bbbb0e72bac0674b1bec3cb09b2f351447b5582386d138431a27e65f6ac
3
- size 46952943
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:07320dccaec6f756afaea092a378331990e411df2fa75158b17c9502cc329cd3
3
+ size 47061419