Update vitGPT.py
Browse files
vitGPT.py
CHANGED
|
@@ -327,7 +327,7 @@ def generate_caption(image,max_tokens=78,temperature=0.9,deterministic=True):
|
|
| 327 |
])
|
| 328 |
|
| 329 |
# image = Image.open(image)
|
| 330 |
-
|
| 331 |
image = gen_tfms(image=image)['image']
|
| 332 |
image = image.unsqueeze(0)
|
| 333 |
sequence = torch.ones(1,1).long() * tokenizer.bos_token_id
|
|
|
|
| 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
|