Spaces:

EngReem85
/

DFU

Sleeping

EngReem85 commited on Jan 5

Commit

42e6af4

verified ·

1 Parent(s): 1a1cddf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ class DenseShuffleGCANet(nn.Module):
 # 3️⃣ تحميل النموذج المدرب
 # -------------------------------
 model = DenseShuffleGCANet(num_classes=4, handcrafted_feature_dim=41)
-model.load_state_dict(torch.load("best_model.pth", map_location=torch.device('cpu')))
 model.eval()
 # -------------------------------

 # 3️⃣ تحميل النموذج المدرب
 # -------------------------------
 model = DenseShuffleGCANet(num_classes=4, handcrafted_feature_dim=41)
+model.load_state_dict(torch.load("best_model_2.pth", map_location=torch.device('cpu')))
 model.eval()
 # -------------------------------