Revert "Use device-agnostic code instead of hardcoded .cuda()"

This reverts commit 4b9afccd53c71186db655e4eb59e0c9c4ee8cc1e.

README.md

@@ -21,20 +21,18 @@ pip install transformers torch torchvision sentencepiece
 ```
 
 ```python
-import torch
 from transformers import AutoModel
 from torchvision import transforms
 from PIL import Image
 import requests
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
 model = AutoModel.from_pretrained("google/tipsv2-l14-dpt", trust_remote_code=True)
-model.eval().to(device)
+model.eval().cuda()
 
 url = "https://raw.githubusercontent.com/google-deepmind/tips/main/scenic/images/example_image_2.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
 transform = transforms.Compose([transforms.Resize((448, 448)), transforms.ToTensor()])
-pixel_values = transform(image).unsqueeze(0).to(device)
+pixel_values = transform(image).unsqueeze(0).cuda()
 
 # All tasks at once
 outputs = model(pixel_values)