Spaces:

Jyothirmai
/

image-captioning-chest-xrays

Sleeping

Jyothirmai commited on Mar 5

Commit

75302e7

•

1 Parent(s): bcd8e41

Update vitGPT.py

Files changed (1) hide show

vitGPT.py CHANGED Viewed

@@ -311,7 +311,7 @@ model_config = SimpleNamespace(
 model = VisionGPT2Model.from_pretrained(model_config)
-model.load_state_dict(torch.load("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
 model.eval()  # Set the model to evaluation mode