Update models.py

models.py

@@ -50,7 +50,7 @@ class VisualFeatureExtractor(nn.Module):
             vit_model = ViTModel(config)
 
             # Load the state dict specifically, excluding 'classifier.bias', 'classifier.weight'
-            state_dict = torch.load('vit-coatten/utils/pytorch_model.bin', map_location=torch.device('cpu'))
+            state_dict = torch.load('pytorch_model.bin', map_location=torch.device('cpu'))
             state_dict = {k: v for k, v in state_dict.items() if not k.startswith('classifier')}
             vit_model.load_state_dict(state_dict, strict=False)