NYUAD-ComNets commited on
Commit
4411f87
·
verified ·
1 Parent(s): 70c43e7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -36,7 +36,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
36
 
37
  model.to(device)
38
 
39
- input_image = Image.open('/home/jovyan/work/image/'+k)
40
 
41
  inputs = processor(text=input_text, images=input_image, padding="longest", do_convert_rgb=True, return_tensors="pt").to(device)
42
  inputs = inputs.to(dtype=model.dtype)
 
36
 
37
  model.to(device)
38
 
39
+ input_image = Image.open(image_path)
40
 
41
  inputs = processor(text=input_text, images=input_image, padding="longest", do_convert_rgb=True, return_tensors="pt").to(device)
42
  inputs = inputs.to(dtype=model.dtype)