Update app.py

app.py

@@ -81,6 +81,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.cpu()
 model.eval()
 
+model.to(device)
+
 import torchvision.transforms as transforms
 def transform_image(image):
     my_transforms = transforms.Compose([transforms.ToTensor(),