Update app.py

app.py

@@ -1,10 +1,26 @@
 import gradio as gr
-from fastai.vision.all import PILImage
-
-from fastai.learner import load_learner
+from fastai.vision.all import *
 
 learn = load_learner('dog_cat_multi.pkl')
 
+cat_breeds = [
+    'Abyssinian', 'Bengal', 'Birman', 'Bombay', 'British_Shorthair',
+    'Egyptian_Mau', 'Main_Coon', 'Persian', 'Ragdoll', 'Russian_Blue',
+    'Siamese', 'Sphynx'
+]
+
+def get_labels(path):
+    regex_pattern = r'(.+)_(\d+)\.jpg$'
+    match = re.match(regex_pattern, str(path.name))
+    if match:
+        label = match.group(1)
+        pet_type = 'cat' if label in cat_breeds else 'dog'
+        return [pet_type, label]
+    else:
+        # Handle the case where the regex doesn't match
+        print(f"Could not match pattern in {path.name}")
+        return []
+
 def predict(image):
     # Transform the image
     pil_image = PILImage.create(image)