Vasudevakrishna commited on
Commit
2d68b5a
·
verified ·
1 Parent(s): 1db1208

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -8,8 +8,8 @@ model, _, transform = open_clip.create_model_and_transforms(
8
  )
9
 
10
  def get_captions(image):
11
- im = Image.open("Car.jpg").convert("RGB")
12
- im = transform(im).unsqueeze(0)
13
 
14
  with torch.no_grad(), torch.cuda.amp.autocast():
15
  generated = model.generate(im)
 
8
  )
9
 
10
  def get_captions(image):
11
+ # im = Image.open("Car.jpg").convert("RGB")
12
+ im = transform(image).unsqueeze(0)
13
 
14
  with torch.no_grad(), torch.cuda.amp.autocast():
15
  generated = model.generate(im)