Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Feb 18, 2024

Commit

5d4c633

verified ·

1 Parent(s): fe70bc2

Update clipGPT.py

Files changed (1) hide show

clipGPT.py CHANGED Viewed

@@ -150,7 +150,7 @@ def generate_caption_clipgpt(img):
     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"
     model = model.to(device)

     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)