tangyuanhao commited on
Commit
d78f40e
·
verified ·
1 Parent(s): 6e8be2e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -14
README.md CHANGED
@@ -12,17 +12,4 @@ cd AFM-Net
12
 
13
  # Usage
14
 
15
- import torch
16
- from your_model_file import AdvancedFusionModel # Make sure to have your model definition available
17
-
18
-
19
-
20
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
21
-
22
-
23
- model = (num_classes=45) # Example for NWPU-RESISC45, adjust as needed
24
-
25
- model_path = "path/to/your/model_weights.pth" # e.g., 'uc_aid_nwpu_weights.pth'
26
- model.load_state_dict(torch.load(model_path, map_location=device))
27
- model.to(device)
28
- model.eval()
 
12
 
13
  # Usage
14
 
15
+ python app.py