Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Feb 18, 2024

Commit

b3556a1

verified ·

1 Parent(s): 76462db

Update clipGPT.py

Files changed (1) hide show

clipGPT.py CHANGED Viewed

@@ -158,8 +158,7 @@ def generate_caption_clipgpt(img):
     tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
     start_time = time.time()
-    image = io.imread(img)
-    pil_image = PIL.Image.fromarray(image)
     image = preprocess(pil_image).unsqueeze(0).to(device)
     with torch.no_grad():

     tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
     start_time = time.time()
     image = preprocess(pil_image).unsqueeze(0).to(device)
     with torch.no_grad():