Spaces:
Sleeping
Sleeping
Commit
·
1efd0a8
1
Parent(s):
8e7a10f
Update app.py
Browse files
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 |
-
|
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 |
|