Sadia commited on
Commit
2e3fc62
·
1 Parent(s): 00320e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -4,15 +4,15 @@ import skimage
4
 
5
  learn = load_learner('curlyhairmodel0.3.pkl')
6
 
7
- categories = ('Straight', 'Curly')
8
 
9
  def predict(img):
10
  img = PILImage.create(img)
11
  pred,pred_idx,probs = learn.predict(img)
12
  return dict(zip(categories, map(float,probs)))
13
 
14
- title = "Pet Breed Classifier"
15
- description = "A pet breed classifier trained on the Oxford Pets dataset with fastai."
16
  examples = [['c1.jpeg'],['c2.jpeg'],['c3.jpeg'],
17
  ['c4.jpeg'],['c5.jpeg'],['c6.jpeg'],
18
  ['c7.jpeg'],['c9.jpeg'],['c10.jpeg'],
 
4
 
5
  learn = load_learner('curlyhairmodel0.3.pkl')
6
 
7
+ categories = ('Curly', 'Straight')
8
 
9
  def predict(img):
10
  img = PILImage.create(img)
11
  pred,pred_idx,probs = learn.predict(img)
12
  return dict(zip(categories, map(float,probs)))
13
 
14
+ title = "Hair Type Classifier"
15
+ description = "A hair type classifier trained on the Oxford Pets dataset with fastai."
16
  examples = [['c1.jpeg'],['c2.jpeg'],['c3.jpeg'],
17
  ['c4.jpeg'],['c5.jpeg'],['c6.jpeg'],
18
  ['c7.jpeg'],['c9.jpeg'],['c10.jpeg'],