Update app.py

app.py

@@ -31,7 +31,7 @@ transform_nor = transforms.Compose([transforms.ToTensor(), transforms.Resize((32
 
 def greet_backdoor(image):
     if image is None:
-        model.load_state_dict(torch.load('./cifar10_badnet.pth'))
+        model.load_state_dict(torch.load('./cifar10_badnet.pth', map_location=torch.device('cpu')))
         return 'changed'
     else:
         image = transform_nor(image).unsqueeze(0)