Upload app.py

app.py

@@ -12,10 +12,10 @@ from glycowork.glycan_data.loader import lib
 def fn(class_list):
   def f(glycan, model):
     if model == "No data augmentation":
-      model = torch.load("model1.pt")
+      model = torch.load("model1.pt", map_location=torch.device('cpu'))
       model.eval()
     else:
-      model = torch.load("model2.pt")
+      model = torch.load("model2.pt", map_location=torch.device('cpu'))
       model.eval()
     glycan = [glycan]
     label = [0]