Update model.py

model.py

@@ -52,7 +52,7 @@ class ViTRecognitionModel(nn.Module):
 
 def load_model(model_path, device='cpu'):
     model = ViTRecognitionModel(vocab_size=vocab_size, hidden_dim=768, max_length=20)
-    model.load_state_dict(torch.load(model_path, map_location=device))
+    model.load_state_dict(torch.load(model_path, map_location=device, weights_only=True))
     model.to(device)
     model.eval()
     return model