Update app.py

app.py

@@ -15,9 +15,9 @@ def predict(input_img):
         drop_rate=0.5,
         num_classes=1,)
     
-    checkpoint1 = torch.load("./machine_full_best.tar")
+    checkpoint1 = torch.load("./machine_full_best.tar",map_location=torch.device('cpu'))
     model1.load_state_dict(checkpoint1['state_dict'])
-    checkpoint2 = torch.load("./human_full_best.tar")
+    checkpoint2 = torch.load("./human_full_best.tar",map_location=torch.device('cpu'))
     model2.load_state_dict(checkpoint2['state_dict'])
     
     my_transform = transforms.Compose([