Update app.py

app.py

@@ -6,7 +6,7 @@ from fastai.tabular.all import *
 from fastai.vision.all import *
 from fastai.vision.utils import get_image_files
 from Ambrosia import pre_process_image
-from huggingface_hub import from_pretrained_fastai, push_to_hub_fastai
+from huggingface_hub import from_pretrained_fastai, push_to_hub_fastai, hf_hub_download
 import gradio as gr
 # Set the token
 os.environ["HUGGINGFACE_TOKEN"] = "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"
@@ -39,7 +39,11 @@ 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 = from_pretrained_fastai(r"ChristopherMarais/Andrew_Alpha_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")