Update app.py

app.py

@@ -7,11 +7,12 @@ import timm
 import pickle
 import torch
 
-#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")
+#   with open('./model.pkl', 'rb') as f:
+#    model = pickle.load(f)
+
+with open('model.pkl', 'rb') as f:
+    model = pickle.load(f)
 
 def is_real(x): return x[0].isupper()