Update app.py

app.py

@@ -1,7 +1,13 @@
 import gradio as gr
 from fastai.vision.all import *
 
-learn = load_learner('model.pkl', cpu=True)
+class CPU_Unpickler(pickle.Unpickler):
+    def find_class(self, module, name):
+        if module == 'torch.storage' and name == '_load_from_bytes':
+            return lambda b: torch.load(io.BytesIO(b), map_location='cpu')
+        else: return super().find_class(module, name)
+
+learn = CPU_Unpickler('model.pkl').load()
 
 labels = learn.dls.vocab
 def predict(img):