Spaces:
Paused
Paused
Update app.py
Browse files
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)
|