work damn you!

app.py

@@ -1,31 +1,42 @@
-
+import torch
+import pickle
+import sys
+import pathlib
 from fastai.vision.all import load_learner, PILImage
-
 import gradio
-import pathlib
-import traceback
-import sys
-
 from gradio import Interface, Image, Label
 
-# Patch pathlib to use WindowsPath on Windows systems
-if pathlib.Path == pathlib.WindowsPath:
-    pathlib.PosixPath = pathlib.WindowsPath
+# Custom Unpickler to handle PosixPath
+
+class CustomUnpickler(pickle.Unpickler):
+    def find_class(self, module, name):
+        if module == "pathlib" and name == "PosixPath":
+            return pathlib.WindowsPath
+        elif module == "pathlib" and name == "WindowsPath":
+            return pathlib.WindowsPath
+        return super().find_class(module, name)
+
 
-def exception_handler(type, value, tb):
-    print("".join(traceback.format_exception(type, value, tb)))
+    def persistent_load(self, pid):
+        return pid
+import dill
+# Load learner using custom unpickler
+def custom_load_learner(fname, cpu=True):
+    map_loc = None if torch.cuda.is_available() and not cpu else 'cpu'
+    with open(fname, 'rb') as f:
+        return dill.load(f)
+from fastai.learner import load_learner
 
-sys.excepthook = exception_handler
 
 # 4. Create a Gradio interface
 def predict_parrot_species(image):
-    learn_inf = load_learner('parrotclassfierwin.pkl')
+    learn_inf = custom_load_learner('parrotclass.pkl')
     pred, _, _ = learn_inf.predict(image)
     return pred
 
 input_image = Image(shape=(224, 224))
-output_label =Label()
+output_label = Label()
 
 gr_interface = gradio.Interface(fn=predict_parrot_species, inputs=input_image, outputs=output_label)
-gr_interface = gradio.Interface(fn=predict_parrot_species, inputs=input_image, outputs=output_label)
-gr_interface.launch()
+gr_interface.launch()
+

parrotclassfierwin.pkl → parrotclass.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ce9f99e1500abd02732fc149f3dfe114befa32e03a031f677d2bb31551fe7868
-size 46985603
+oid sha256:6657a442bfa942517a8f2d9f03a768fdf6f9695c12eb47b39140cde7a71d8592
+size 46973589