wnagleiofficial commited on
Commit
7bf7082
·
1 Parent(s): 906530e
Files changed (1) hide show
  1. NeuroPredPLM/predict.py +1 -1
NeuroPredPLM/predict.py CHANGED
@@ -13,6 +13,6 @@ def predict(peptide_list, model_path, device='cpu'):
13
  prob, att = neuroPred_model(peptide_list, device)
14
  pred = torch.softmax(prob, dim=-1).cpu().tolist()
15
  att = att.cpu().numpy()
16
- out = {'Neuroppetide':pred[0][1], "Non-neuropeptide":pred[0][0]}
17
  return out
18
 
 
13
  prob, att = neuroPred_model(peptide_list, device)
14
  pred = torch.softmax(prob, dim=-1).cpu().tolist()
15
  att = att.cpu().numpy()
16
+ out = {'Neuropeptide':pred[0][1], "Non-neuropeptide":pred[0][0]}
17
  return out
18