Update app.py

app.py

@@ -1,8 +1,21 @@
 from fastai.vision.all import *
 import gradio as gr
 
+path = untar_data(URLs.PETS)
+
 learn = load_learner('export.pkl')
 
+files = get_image_files(path/'images')
+
+pat = r'^(.*)_\d+.jpg'
+
+dls = ImageDataLoaders.from_name_re(path, files, pat, item_tfms=Resize(460),
+                                    batch_tfms=aug_transforms(size=224))
+
+learn = vision_learner(dls, resnet34, metrics=error_rate)
+
+learn.fine_tune(2, 0.0008)
+
 labels = learn.dls.vocab
 def predict(img):
     img = PILImage.create(img)