ChristopherMarais commited on
Commit
1efd0a8
·
1 Parent(s): 8e7a10f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -39,9 +39,7 @@ def unkown_prob_calc(probs, wedge_threshold, wedge_magnitude=1, wedge='strict'):
39
  return(unknown_prob)
40
 
41
  # load model
42
- # learn = load_learner("E:\\GIT_REPOS\\Beetle_classifier\\Models\\beetle_classifier.pkl", cpu=False)
43
- # learn = load_learner(r"beetle_classifier.pkl", cpu=True)
44
- learn = from_pretrained_fastai("ChristopherMarais/Andrew_Alpha_model")
45
  # get class names
46
  labels = np.append(np.array(learn.dls.vocab), "Unknown")
47
 
 
39
  return(unknown_prob)
40
 
41
  # load model
42
+ learn = from_pretrained_fastai(r"ChristopherMarais/Andrew_Alpha_model")
 
 
43
  # get class names
44
  labels = np.append(np.array(learn.dls.vocab), "Unknown")
45