wnagleiofficial commited on
Commit
906530e
·
1 Parent(s): 3ba36c5
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ def classifier(peptide_seq):
10
  for record in SeqIO.parse(handle, 'fasta'):
11
  data.append((record.id, record.seq))
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
- neuropeptide_pred = predict(data,device)
14
  return neuropeptide_pred
15
  # {peptide_id:[Type:int(1->neuropeptide,0->non-neuropeptide), attention score:nd.array]}
16
 
 
10
  for record in SeqIO.parse(handle, 'fasta'):
11
  data.append((record.id, record.seq))
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
+ neuropeptide_pred = predict(data, './model.pth', device)
14
  return neuropeptide_pred
15
  # {peptide_id:[Type:int(1->neuropeptide,0->non-neuropeptide), attention score:nd.array]}
16