Update app.py

app.py

@@ -5,9 +5,8 @@ import torchvision.transforms as transforms
 
 model = torchvision.models.resnet50(pretrained=True)
 model.fc = nn.Linear(model.fc.in_features, 2)
-model.load_state_dict(torch.load("model.pth", map_location=torch.device('cpu')))
+model.load_state_dict(torch.load("model.pth"))
 model.eval()
-device = torch.device("cpu")
 
 transform = transforms.Compose([
     transforms.Resize((224, 224)),