Spaces:

oucgc1996
/

ConotoxinFinder

Sleeping

oucgc1996 commited on Jul 26

Commit

cf0d6b6

•

1 Parent(s): 1a8df8b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ class MyModel(nn.Module):
         return torch.softmax(output_feature,dim=1)
 model = MyModel()
-model.load_state_dict(torch.load("best_model.pth", map_location=torch.device('cpu')))
 model = model.to(device)
 model.eval()

         return torch.softmax(output_feature,dim=1)
 model = MyModel()
+model.load_state_dict(torch.load("best_model.pth", map_location=torch.device('cpu')), strict=False)
 model = model.to(device)
 model.eval()