EngReem85 commited on
Commit
f52464c
·
verified ·
1 Parent(s): 47533fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
17
  # تحميل النموذج
18
  model = DenseShuffleGCANet(num_classes=4, handcrafted_feature_dim=41)
19
  model.load_state_dict(
20
- torch.load("best_model.pth", map_location=device)
21
  )
22
  model.to(device)
23
  model.eval()
 
17
  # تحميل النموذج
18
  model = DenseShuffleGCANet(num_classes=4, handcrafted_feature_dim=41)
19
  model.load_state_dict(
20
+ torch.load("best_model_2.pth", map_location=device)
21
  )
22
  model.to(device)
23
  model.eval()