Bhadralab commited on
Commit
34fd3ae
·
verified ·
1 Parent(s): ef06ce0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def read_fasta(fasta_string):
38
 
39
  def predict_peptide_class(sequences):
40
  """Predicts peptide classifications in batch."""
41
- inputs = tokenizer(sequences, return_tensors='pt', padding=True, truncation=True)
42
  with torch.no_grad():
43
  logits = model(**inputs).logits
44
  probs = torch.sigmoid(logits)[:, 1].cpu().numpy()
 
38
 
39
  def predict_peptide_class(sequences):
40
  """Predicts peptide classifications in batch."""
41
+ inputs = tokenizer(sequences, return_tensors='pt')
42
  with torch.no_grad():
43
  logits = model(**inputs).logits
44
  probs = torch.sigmoid(logits)[:, 1].cpu().numpy()