Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Feb 17, 2024

Commit

69fe50a

verified ·

1 Parent(s): 3037cd2

Update clipGPT.py

Files changed (1) hide show

clipGPT.py CHANGED Viewed

@@ -134,7 +134,7 @@ def generate_beam(
 def generate_caption_clipgpt(img):
     prefix_length = 10
-    model = ClipGPT2Model(prefix_length, img_feature_size = feature_dim)
     model.load_state_dict(torch.load('model_train_best_run_clipGPT.pt'))
     model = model.eval()
     device = "cuda" if torch.cuda.is_available() else "cpu"

 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'))
     model = model.eval()
     device = "cuda" if torch.cuda.is_available() else "cpu"