Spaces:

oucgc1996
/

Antimicrobial-peptide-generation

Runtime error

oucgc1996 commited on Feb 12, 2023

Commit

38c73f6

•

1 Parent(s): cfed6a2

app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -42,7 +42,7 @@ def AMP(file):
             return torch.softmax(output_feature,dim=1)
     model = MyModel()
-    model.load_state_dict(torch.load("AMP_final_best_b256_Ir1e5_esm_0.pth",map_location=torch.device('cpu')))
     model = model.to(device)
     model.eval()
     out_probability = []

             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()
     out_probability = []