Update README.md

README.md

@@ -47,13 +47,12 @@ from io import BytesIO
 from transformers import AutoProcessor, AutoModelForVision2Seq
 from transformers.image_utils import load_image
 
-DEVICE = "cuda:0"
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Note that passing the image urls (instead of the actual pil images) to the processor is also possible
 image = load_image("https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg")
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
-
 model_id = "cmarkea/idefics2-8b-ft-docvqa-lora"
 processor = AutoProcessor.from_pretrained("HuggingFaceM4/idefics2-8b", do_image_splitting=False)
 model = AutoModelForVision2Seq.from_pretrained(