Update app.py

app.py

@@ -50,8 +50,8 @@ def predict(img,text):
 
   image_features.cpu().numpy()
 
-  image_mean = torch.tensor([0.48145466, 0.4578275, 0.40821073]).cuda()
-  image_std = torch.tensor([0.26862954, 0.26130258, 0.27577711]).cuda()
+  image_mean = torch.tensor([0.48145466, 0.4578275, 0.40821073])
+  image_std = torch.tensor([0.26862954, 0.26130258, 0.27577711])
 
   images = [preprocess(im) for im in l]
   image_input = torch.tensor(np.stack(images)).cuda()