vizcodes commited on
Commit
45e82fb
·
1 Parent(s): fa319c4

changed path to model pickle file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from fastai.learner import load_learner
4
 
5
  def image_classifier(inp):
6
 
7
- art_classifier = load_learner('/home/vizcodes/projects/fast-ai/testing/art_classifier.pkl')
8
  class_lst = art_classifier.dls.vocab
9
  pred,pred_idx,probs = art_classifier.predict(inp)
10
  pred_dict = {x:y.numpy() for x, y in zip(class_lst,probs)}
 
4
 
5
  def image_classifier(inp):
6
 
7
+ art_classifier = load_learner('art_classifier.pkl')
8
  class_lst = art_classifier.dls.vocab
9
  pred,pred_idx,probs = art_classifier.predict(inp)
10
  pred_dict = {x:y.numpy() for x, y in zip(class_lst,probs)}