doyu commited on
Commit
84d5588
1 Parent(s): 72649b5
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,9 +1,10 @@
1
  from fastai.vision.all import *
2
  import gradio as gr
3
 
 
4
  learn = load_learner('model.pkl')
5
 
6
- categories = ('female', 'male')
7
 
8
  def classify_image(img):
9
  pred,idx,probs = learn.predict(img)
 
1
  from fastai.vision.all import *
2
  import gradio as gr
3
 
4
+ def is_cat(x): return x[0].isupper()
5
  learn = load_learner('model.pkl')
6
 
7
+ categories = ('dog', 'cat')
8
 
9
  def classify_image(img):
10
  pred,idx,probs = learn.predict(img)