Update vitGPT.py
Browse files
vitGPT.py
CHANGED
|
@@ -326,8 +326,8 @@ def generate_caption(image,max_tokens=50,temperature=0.9,deterministic=True):
|
|
| 326 |
ToTensorV2()
|
| 327 |
])
|
| 328 |
|
| 329 |
-
image = Image.open(image)
|
| 330 |
-
image = np.array(image)
|
| 331 |
image = gen_tfms(image=image)['image']
|
| 332 |
image = image.unsqueeze(0)
|
| 333 |
sequence = torch.ones(1,1).long() * tokenizer.bos_token_id
|
|
|
|
| 326 |
ToTensorV2()
|
| 327 |
])
|
| 328 |
|
| 329 |
+
# image = Image.open(image)
|
| 330 |
+
# image = np.array(image)
|
| 331 |
image = gen_tfms(image=image)['image']
|
| 332 |
image = image.unsqueeze(0)
|
| 333 |
sequence = torch.ones(1,1).long() * tokenizer.bos_token_id
|