Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Mar 6

Commit

8b9889c

•

1 Parent(s): e6a32cf

Update vitGPT.py

Files changed (1) hide show

vitGPT.py CHANGED Viewed

@@ -316,7 +316,7 @@ print(model)
 model.eval()  # Set the model to evaluation mode
-def generate_caption(image,max_tokens=78,temperature=0.9,deterministic=True):
     tokenizer = GPT2TokenizerFast.from_pretrained('gpt2')
     tokenizer.pad_token = tokenizer.eos_token

 model.eval()  # Set the model to evaluation mode
+def generate_caption(image,max_tokens,temperature,deterministic=True):
     tokenizer = GPT2TokenizerFast.from_pretrained('gpt2')
     tokenizer.pad_token = tokenizer.eos_token