Update app.py

app.py

@@ -39,10 +39,10 @@ def unkown_prob_calc(probs, wedge_threshold, wedge_magnitude=1, wedge='strict'):
     return(unknown_prob)
 
 # load model
-# learn = from_pretrained_fastai(r"ChristopherMarais/Andrew_Alpha_model")
-learn = load_learner(
-    hf_hub_download('ChristopherMarais/Andrew_Alpha_model', filename="model.pkl")
-    )
+learn = from_pretrained_fastai("ChristopherMarais/beetle-model")
+# learn = load_learner(
+#     hf_hub_download('ChristopherMarais/Andrew_Alpha_model', filename="model.pkl")
+#     )
     
 # get class names
 labels = np.append(np.array(learn.dls.vocab), "Unknown")