george sami commited on
Commit
c85f5c7
·
1 Parent(s): f59a10d

add packgaes to requirements

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -4,6 +4,8 @@ from fastai.vision.all import *
4
  learner = load_learner("model.pkl")
5
  categories = ("cat", "dog")
6
 
 
 
7
  def classify_image(img):
8
  pred, pred_idx, probs = learner.predict(img)
9
  return {categories[i]: float(probs[i]) for i in range(len(categories))}
 
4
  learner = load_learner("model.pkl")
5
  categories = ("cat", "dog")
6
 
7
+ def is_cat(x): return x[0].isupper()
8
+
9
  def classify_image(img):
10
  pred, pred_idx, probs = learner.predict(img)
11
  return {categories[i]: float(probs[i]) for i in range(len(categories))}