updated gradio and compatibility with image types

app.py

@@ -1,11 +1,16 @@
 from fastai.vision.all import load_learner
+from PIL import Image
 import gradio as gr
-# import pathlib
+import pathlib
+from pillow_heif import register_heif_opener
+import pillow_avif
+
+register_heif_opener()
 
 # For the posix path error: when you train your model on colab/gradient and download it, then do inference on Windows.
 # Redirect PosixPath to WindowsPath:
-# temp = pathlib.PosixPath
-# pathlib.PosixPath = pathlib.WindowsPath
+temp = pathlib.PosixPath
+pathlib.PosixPath = pathlib.WindowsPath
 
 
 # Data below sourced from:
@@ -14,12 +19,15 @@ import gradio as gr
 dog_breeds = tuple(['affenpinscher', 'afghan hound', 'african', 'airedale', 'akita', 'american terrier', 'appenzeller', 'australian cattledog', 'australian terrier', 'basenji', 'basset hound', 'beagle', 'bedlington terrier', 'bernese mountain', 'bichon frise', 'blenheim spaniel', 'blood hound', 'bluetick', 'border collie', 'border terrier', 'borzoi', 'boston bulldog', 'bouvier', 'boxer', 'brabancon', 'briard', 'brittany spaniel', 'bull mastiff', 'cairn terrier', 'cardigan corgi', 'caucasian ovcharka', 'cavapoo', 'chesapeake retriever', 'chihuahua', 'chow', 'clumber', 'cockapoo', 'cocker spaniel', 'coonhound', 'cotondetulear', 'curly retriever', 'dachshund', 'dalmatian', 'dandie terrier', 'dhole', 'dingo', 'doberman', 'english bulldog', 'english hound', 'english mastiff', 'english setter', 'english sheepdog', 'english springer', 'entlebucher', 'eskimo', 'flatcoated retriever', 'fox terrier', 'french bulldog', 'german pointer', 'germanlonghair pointer', 'germanshepherd', 'giant schnauzer', 'golden retriever', 'gordon setter', 'great dane', 'groenendael', 'havanese', 'husky', 'ibizan hound', 'irish setter', 'irish spaniel', 'irish terrier', 'irish wolfhound', 'italian greyhound', 'italian segugio', 'japanese spaniel', 'japanese spitz', 'keeshond', 'kelpie', 'kerryblue terrier', 'komondor', 'kuvasz', 'labradoodle', 'labrador', 'lakeland terrier', 'lapphund finnish', 'leonberg', 'lhasa', 'malamute', 'malinois', 'maltese', 'medium poodle', 'mexicanhairless', 'miniature pinscher', 'miniature poodle', 'miniature schnauzer', 'mix', 'newfoundland', 'norfolk terrier', 'norwegian buhund', 'norwegian elkhound', 'norwich terrier', 'otterhound', 'papillon', 'patterdale terrier', 'pekinese', 'pembroke', 'pitbull', 'plott hound', 'pomeranian', 'pug', 'puggle', 'pyrenees', 'redbone', 'rhodesian ridgeback', 'rottweiler', 'russell terrier', 'saluki', 'samoyed', 'schipperke', 'scottish deerhound', 'scottish terrier', 'sealyham terrier', 'sharpei', 'shepherd australian', 'shetland sheepdog', 'shiba', 'shihtzu', 'silky terrier', 'spanish waterdog', 'staffordshire bullterrier', 'standard poodle', 'stbernard', 'sussex spaniel', 'swiss mountain', 'tervuren', 'tibetan mastiff', 'tibetan terrier', 'toy poodle', 'toy terrier', 'vizsla', 'walker hound', 'weimaraner', 'welsh spaniel', 'welsh terrier', 'westhighland terrier', 'wheaten terrier', 'whippet', 'yorkshire terrier'])
 
 def classify_image(img):
-    _, _, probs = learn.predict(img)
-    return dict(zip(dog_breeds, map(float, probs)))
+    try:
+        _, _, probs = learn.predict(img)
+        return dict(zip(dog_breeds, map(float, probs)))
+    except Exception as e:
+        raise gr.Error("Invalid Image Input Type")
 
 learn = load_learner('dogIdentifierModel.pkl')
 
-image = gr.components.Image(shape=(396, 396))
+image = gr.components.Image(image_mode='RGB')
 label = gr.components.Label()
 examples = ['golden-retriever.jpg', 'german-shepherd.jpg', 'doberman.jpg', 'husky.jpg']
 

requirements.txt

@@ -2,8 +2,10 @@ matplotlib==3.5.2
 timm==0.9.7
 fastai==2.7.12
 torch==2.0.0
-gradio==3.1.4
+gradio==4.1.1
 numpy==1.23.5
 pandas==2.0.2
 Pillow==9.2.0
-uvicorn==0.18.2
+uvicorn==0.18.2
+pillow-avif-plugin==1.4.1
+pillow-heif==0.13.1