Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Mar 4

Commit

748c95d

•

1 Parent(s): 247c99a

Update clipGPT.py

Files changed (1) hide show

clipGPT.py CHANGED Viewed

@@ -159,7 +159,7 @@ def generate_caption_clipgpt(img):
     start_time = time.time()
     # pil_image = PIL.Image.fromarray(img)
-    image = preprocess(pil_image).unsqueeze(0).to(device)
     with torch.no_grad():
         prefix = clip_model.encode_image(image).to(device, dtype=torch.float32)

     start_time = time.time()
     # pil_image = PIL.Image.fromarray(img)
+    image = preprocess(img).unsqueeze(0).to(device)
     with torch.no_grad():
         prefix = clip_model.encode_image(image).to(device, dtype=torch.float32)