marufc36 commited on
Commit
d37ff56
·
1 Parent(s): 6446ee9

update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ cap_labels=(
13
  'trucker cap'
14
 
15
  )
16
- model=load_learner(f'models/cap-classifier-v2.pkl')
17
 
18
  def recognize_image(image):
19
  pred, idx, probs=model.predict(image)
 
13
  'trucker cap'
14
 
15
  )
16
+ model=load_learner(f'cap-classifier-v2.pkl')
17
 
18
  def recognize_image(image):
19
  pred, idx, probs=model.predict(image)