yay!

app.py

@@ -2,23 +2,17 @@ import torch
 import sys
 import pathlib
 from pathlib import Path
-from fastai.vision.all import *
-from torchvision.models import resnet18
-import gradio
-from gradio import Interface, Image, Label
-import torch
-import sys
-import pathlib
-from pathlib import Path
 from fastai.vision.all import load_learner, PILImage
 import gradio
 from gradio import Interface, Image, Label
+
 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(str(Path('parrotclass.pkl')))
+    model_path = Path('parrotclass.pkl').resolve()  # Resolve the path object
+    learn_inf = load_learner(str(model_path))  # Convert the Path object to a string
     pred, _, _ = learn_inf.predict(image)
     return pred