Himetsu commited on
Commit
4d53906
1 Parent(s): 8dc42df

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -14,10 +14,10 @@ model = LlavaForConditionalGeneration.from_pretrained(model_id, load_in_4bit=Tru
14
  processor = AutoProcessor.from_pretrained(model_id)
15
 
16
  IMG_URLS = [
17
- Image.open(requests.get("https://picsum.photos/id/237/400/300", stream=True).raw),
18
- Image.open(requests.get("https://picsum.photos/id/231/200/300", stream=True).raw),
19
- Image.open(requests.get("https://picsum.photos/id/27/500/500", stream=True).raw),
20
- Image.open(requests.get("https://picsum.photos/id/17/150/600", stream=True).raw),
21
  ]
22
  PROMPT = "<s>[INST]Describe the images.\n[IMG][IMG][IMG][IMG][/INST]"
23
 
 
14
  processor = AutoProcessor.from_pretrained(model_id)
15
 
16
  IMG_URLS = [
17
+ "https://picsum.photos/id/237/400/300",
18
+ "https://picsum.photos/id/231/200/300",
19
+ "https://picsum.photos/id/27/500/500",
20
+ "https://picsum.photos/id/17/150/600",
21
  ]
22
  PROMPT = "<s>[INST]Describe the images.\n[IMG][IMG][IMG][IMG][/INST]"
23