Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Feb 18, 2024

Commit

2f1cbf2

verified ·

1 Parent(s): 5d4c633

Update clipGPT.py

Files changed (1) hide show

clipGPT.py CHANGED Viewed

@@ -148,7 +148,7 @@ 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=torch.device('cpu')
     model = model.to(device)

     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')
     model = model.to(device)