Spaces:
Runtime error
Runtime error
wnagleiofficial
commited on
Commit
·
7bf7082
1
Parent(s):
906530e
fix typo
Browse files- 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 = {'
|
| 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 |
|