Update model.py

model.py

@@ -52,6 +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)
+    # Set weights_only=True to address the FutureWarning
     model.load_state_dict(torch.load(model_path, map_location=device, weights_only=True))
     model.to(device)
     model.eval()