Update app.py

app.py

@@ -157,16 +157,15 @@ def compute_plddt_iptm(protein_seq, binder_seq):
         outputs = model(**inputs)
         plddt = outputs.plddt
         
-        ptm = outputs.ptm
+        ptm = outputs.ptm.item()
         print(f'161: ptm: {ptm:.4f}')
-        ptm2 = outputs["ptm"]
-        print("163: ptm2:",ptm2)
+
         
     avg_plddt = plddt[0,:,1].mean()
     print("166: avg_plddt:",avg_plddt)
     plddt2 = outputs["plddt"][0,:,1].mean()
     print("169: plddt2:",plddt2)
-    iPTM = ptm.mean()
+    iPTM = {ptm.mean():%.4f}
     print("170: iPTM:",iPTM)
     
     return avg_plddt, iPTM
@@ -225,7 +224,7 @@ def predict_peptide(base_model_path, finetuned_model_path, input_seqs, peptide_l
 
     if isinstance(input_seqs, str):  # Single sequence
         binders = generate_peptide_for_single_sequence(loaded_model, tokenizer, input_seqs, peptide_length, top_k, num_binders)
-        results_df = pd.DataFrame(binders, columns=['Binder', 'Pseudo Perplexity', 'pLDDT', 'iPTM'])
+        results_df = pd.DataFrame(binders, columns=['Binder', 'PPL', 'pLDDT', 'iPTM'])
 
     elif isinstance(input_seqs, list):  # List of sequences
         results = []