Load the model from huggingface

app.py

@@ -1,12 +1,16 @@
 import gradio as gr
 from fastai.learner import load_learner
 from fastai.vision.core import PILImage
+from huggingface_hub import hb_hub_download
+
+learner = load_learner(hb_hub_download("pinkpekoe/lesson2-bear-classifier", "export.pkl"))
+
 
 learner = load_learner('export.pkl')
 
 def predict_bear_type(img_path):
     img = PILImage.create(img_path)
-    pred,pred_idx,probs = learner.predict(img)
+    pred, pred_idx, probs = learner.predict(img)
     return f"Prediction: {pred}; Probability: {probs[pred_idx]:.02f}"
 
 iface = gr.Interface(fn=predict_bear_type, inputs="image", outputs="text")

requirements.txt

@@ -1 +1,2 @@
 fastai
+huggingface_hub