Update vitGPT.py

vitGPT.py

@@ -315,7 +315,7 @@ model.load_state_dict(torch.load("captioner.pt", map_location='cpu')) # Use 'cud
 model.eval()  # Set the model to evaluation mode
 
 
-def generate_caption(image,max_tokens=50,temperature=0,deterministic=True):
+def generate_caption(image,max_tokens=78,temperature=0.9,deterministic=True):
     tokenizer = GPT2TokenizerFast.from_pretrained('gpt2')
     tokenizer.pad_token = tokenizer.eos_token