Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Mar 5

Commit

538ab5e

•

1 Parent(s): 4c10f74

Update vitGPT.py

Files changed (1) hide show

vitGPT.py CHANGED Viewed

@@ -312,6 +312,7 @@ model_config = SimpleNamespace(
 model = VisionGPT2Model.from_pretrained(model_config)
 model.load_state_dict(torch.load("model/captioner.pt", map_location='cpu')) # Use 'cuda' if you have a GPU
 model.eval()  # Set the model to evaluation mode

 model = VisionGPT2Model.from_pretrained(model_config)
 model.load_state_dict(torch.load("model/captioner.pt", map_location='cpu')) # Use 'cuda' if you have a GPU
+print(model)
 model.eval()  # Set the model to evaluation mode