skcoirz commited on
Commit
14d8f2b
Β·
1 Parent(s): c25c5bc

add is_cat

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import gradio as gr
4
  learner = load_learner('model.pkl')
5
  categories = ("Dog", "Cat")
6
 
7
-
8
  def infer(img):
9
  _, _, prob = learner.predict(img)
10
  return dict(zip(categories, map(float, prob)))
 
4
  learner = load_learner('model.pkl')
5
  categories = ("Dog", "Cat")
6
 
7
+ def is_cat(x): return x[0].isupper()
8
  def infer(img):
9
  _, _, prob = learner.predict(img)
10
  return dict(zip(categories, map(float, prob)))