import torch
from your_model_file import AdvancedFusionModel # Make sure to have your model definition available

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

model = (num_classes=45) # Example for NWPU-RESISC45, adjust as needed

model_path = "path/to/your/model_weights.pth" # e.g., 'uc_aid_nwpu_weights.pth'
model.load_state_dict(torch.load(model_path, map_location=device))
model.to(device)
model.eval()