Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Feb 18

Commit

37335e4

•

1 Parent(s): 6ae4f98

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(image)
     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():