yay!

app.py

@@ -4,14 +4,21 @@ import pathlib
 from fastai.vision.all import load_learner, PILImage
 import gradio
 from gradio import Interface, Image, Label
+import logging
+logging.basicConfig(level=logging.INFO)
 def custom_resnet_splitter(model):
     resnet = model[0]
     return [params(resnet[0]), params(resnet[1]), params(resnet[4]), params(resnet[5]), params(resnet[6]), params(resnet[7]), params(model[1])]
 
 def predict_parrot_species(image):
-    learn_inf = load_learner('parrotclass.pkl')
-    pred, _, _ = learn_inf.predict(image)
-    return pred
+    try:
+        learn_inf = load_learner('parrotclass.pkl')
+        pred, _, _ = learn_inf.predict(image)
+        return pred
+    except Exception as e:
+        logging.error(f"Error during inference: {e}")
+        return "Error during inference"
+
 
 input_image = Image(shape=(224, 224))
 output_label = Label()

requirements.txt

@@ -1,2 +1,4 @@
-fastai==2.6.1
-fastcore
+fastai
+fastcore
+torch
+torchvision