Update app.py

app.py

@@ -5,9 +5,13 @@ from fastai.vision.all import *
 import gradio as gr
 import timm
 import pickle
+import torch
 
-with open('./model.pkl', 'rb') as f:
-    model = pickle.load(f)
+#with open("model.pkl", "rb") as f:
+#    model = torch.load(f, map_location="cpu")
+
+with open("model.pkl", "rb") as f:
+    model = torch.load(f, map_location="cpu")
 
 def is_real(x): return x[0].isupper()