Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Mar 4

Commit

247c99a

•

1 Parent(s): 281dd11

Update clipGPT.py

Files changed (1) hide show

clipGPT.py CHANGED Viewed

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

     tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
     start_time = time.time()
+    # pil_image = PIL.Image.fromarray(img)
     image = preprocess(pil_image).unsqueeze(0).to(device)
     with torch.no_grad():