fix app.py

app.py

@@ -1,15 +1,16 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: app.ipynb.
 
 # %% auto 0
-__all__ = ['labels', 'title', 'description', 'article', 'image', 'label', 'examples', 'interpretation', 'enable_queue', 'intf',
-           'classify_image']
+__all__ = ['learner', 'labels', 'title', 'description', 'article', 'image', 'label', 'examples', 'interpretation', 'enable_queue',
+           'intf', 'classify_image']
 
 # %% app.ipynb 1
 from fastai.vision.all import *
 import scipy
 import gradio as gr
 
-# %% app.ipynb 3
+# %% app.ipynb 2
+learner = load_learner('pet_learner.pkl')
 labels = learner.dls.vocab
 
 def classify_image(img):
@@ -20,7 +21,7 @@ def classify_image(img):
     pred, pred_idx, probs = learner.predict(img)
     return dict(zip(labels, map(float, probs)))
 
-# %% app.ipynb 4
+# %% app.ipynb 3
 title = "Cat & Dog Breed Classifier"
 description = "A pet breed classifier trained on the Oxford Pets dataset with fastai. Created as a demo for Gradio and HuggingFace Spaces."
 article="<p style='text-align: center'><a href='https://tmabraham.github.io/blog/gradio_hf_spaces_tutorial' target='_blank'>Blog post</a></p>"
@@ -40,4 +41,3 @@ intf = gr.Interface(classify_image,
                     interpretation=interpretation)
 
 intf.launch(enable_queue=enable_queue)
-