Update vitGPT.py

vitGPT.py

@@ -326,8 +326,8 @@ def generate_caption(image,max_tokens=50,temperature=0.9,deterministic=True):
             ToTensorV2()
         ])
 
-    image = Image.open(image)
-    image = np.array(image)
+    # image = Image.open(image)
+    # image = np.array(image)
     image = gen_tfms(image=image)['image']
     image = image.unsqueeze(0)
     sequence = torch.ones(1,1).long() * tokenizer.bos_token_id