AylinM commited on
Commit
e280d25
1 Parent(s): 9cecc3b

remove duplicated code

Browse files
Files changed (1) hide show
  1. app.py +0 -7
app.py CHANGED
@@ -8,13 +8,6 @@ def get_number_of_cats(file_name):
8
 
9
  learn = load_learner('./cat_counter.pkl')
10
 
11
- mapping = {'cat_counter/cat': 1.0, 'cat_counter/two cats': 2.0, 'cat_counter/three cats': 3.0}
12
-
13
- def get_number_of_cats(file_name):
14
- return mapping.get(str(file_name.parent), 0)
15
-
16
- learn = load_learner('cat_counter.pkl')
17
-
18
  def predict(img):
19
  img = PILImage.create(img)
20
  pred,pred_idx,probs = learn.predict(img)
 
8
 
9
  learn = load_learner('./cat_counter.pkl')
10
 
 
 
 
 
 
 
 
11
  def predict(img):
12
  img = PILImage.create(img)
13
  pred,pred_idx,probs = learn.predict(img)