Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -346,8 +346,9 @@ def predict_peptide_from_file(base_model_path, finetuned_model_path, file_obj, m
|
|
| 346 |
|
| 347 |
#get metrics for ground truth peptide
|
| 348 |
ppl = compute_pseudo_perplexity(loaded_model, tokenizer, protein_seq, peptide_seq)
|
| 349 |
-
if plddt_iptm_yes=="yes":
|
| 350 |
-
plddt, iptm =
|
|
|
|
| 351 |
else:
|
| 352 |
plddt, iptm = [0, 0]
|
| 353 |
|
|
|
|
| 346 |
|
| 347 |
#get metrics for ground truth peptide
|
| 348 |
ppl = compute_pseudo_perplexity(loaded_model, tokenizer, protein_seq, peptide_seq)
|
| 349 |
+
if plddt_iptm_yes=="yes":
|
| 350 |
+
plddt, iptm = get_plddt(protein_seq, generated_binder)
|
| 351 |
+
#plddt, iptm = compute_plddt_iptm(protein_seq, peptide_seq) #too time-consuming
|
| 352 |
else:
|
| 353 |
plddt, iptm = [0, 0]
|
| 354 |
|