wangjin2000 commited on
Commit
3936a88
·
verified ·
1 Parent(s): c1373de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -162,7 +162,7 @@ def generate_peptide_for_single_sequence(model, tokenizer, protein_seq, peptide_
162
  input_sequence = protein_seq + masked_peptide
163
  inputs = tokenizer(input_sequence, return_tensors="pt").to(model.device)
164
  print("164: inputs:",inputs)
165
- print("165: inputs.shape:",inputs.size)
166
  with torch.no_grad():
167
  logits = model(**inputs).logits
168
  print("166: logits:", logits)
 
162
  input_sequence = protein_seq + masked_peptide
163
  inputs = tokenizer(input_sequence, return_tensors="pt").to(model.device)
164
  print("164: inputs:",inputs)
165
+ #print("165: inputs.shape:",inputs.size)
166
  with torch.no_grad():
167
  logits = model(**inputs).logits
168
  print("166: logits:", logits)