Update README.md

README.md

@@ -11,7 +11,7 @@ from transformers import AutoModelForVision2Seq, AutoProcessor
 from PIL import Image
 import torch
 device = "cuda:2"
-repo = "kosmos2_5"
+repo = "kirp/kosmos2_5"
 dtype = torch.float16
 # dtype = torch.bfloat16
 model = AutoModelForVision2Seq.from_pretrained(repo, device_map = device, torch_dtype=dtype)