Update app.py

app.py

@@ -54,7 +54,7 @@ def predict(img,text):
   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()
+  image_input = torch.tensor(np.stack(images))
   image_input -= image_mean[:, None, None]
   image_input /= image_std[:, None, None]
   with torch.no_grad():