Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -122,6 +122,8 @@ def finetune(base_model_path, peptide_length): #, train_dataset, test_dataset)
|
|
| 122 |
def compute_pseudo_perplexity(model, tokenizer, protein_seq, binder_seq):
|
| 123 |
sequence = protein_seq + binder_seq
|
| 124 |
original_input = tokenizer.encode(sequence, return_tensors='pt').to(model.device)
|
|
|
|
|
|
|
| 125 |
length_of_binder = len(binder_seq)
|
| 126 |
|
| 127 |
# Prepare a batch with each row having one masked token from the binder sequence
|
|
|
|
| 122 |
def compute_pseudo_perplexity(model, tokenizer, protein_seq, binder_seq):
|
| 123 |
sequence = protein_seq + binder_seq
|
| 124 |
original_input = tokenizer.encode(sequence, return_tensors='pt').to(model.device)
|
| 125 |
+
print("125: original_input:", original_input)
|
| 126 |
+
print("126: original_input.size:", original_input.shape)
|
| 127 |
length_of_binder = len(binder_seq)
|
| 128 |
|
| 129 |
# Prepare a batch with each row having one masked token from the binder sequence
|