Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Feb 18

Commit

65044cf

•

1 Parent(s): 3818998

Update clipGPT.py

Files changed (1) hide show

clipGPT.py CHANGED Viewed

@@ -147,7 +147,7 @@ def generate_beam(
 def generate_caption_clipgpt(img):
     prefix_length = 10
-    model = ClipGPT2Model(prefix_length, img_feature_size = 512)
     model.load_state_dict(torch.load('model_train_best_run_clipGPT.pt', map_location=torch.device('cpu')))
     model = model.eval()
     device=torch.device('cpu')

 def generate_caption_clipgpt(img):
     prefix_length = 10
+    model = ClipGPT2Model(prefix_length)
     model.load_state_dict(torch.load('model_train_best_run_clipGPT.pt', map_location=torch.device('cpu')))
     model = model.eval()
     device=torch.device('cpu')