Update app.py

app.py

@@ -261,13 +261,13 @@ def predict_peptide_from_file(base_model_path, finetuned_model_path, file_obj, m
         ppl_value = compute_pseudo_perplexity(loaded_model, tokenizer, protein_seq, peptide_seq)
         plddt_value, iPTM_value = compute_plddt_iptm(protein_seq, peptide_seq)
 
-        results.append([seq, binder, ppl, plddt, iptm, 1]) # flag 1 for ground truth peptide
+        results.append([protein_seq, binder, ppl, plddt, iptm, 1]) # flag 1 for ground truth peptide
 
         #predict peptides
         binders = generate_peptide_for_single_sequence(loaded_model, tokenizer, protein_seq, peptide_length, top_k, num_binders)
         
         for binder, ppl, plddt, iptm in binders:    
-            results.append([seq, binder, ppl, plddt, iptm, 0]) # flag 0 for generated peptide
+            results.append([protein_seq, binder, ppl, plddt, iptm, 0]) # flag 0 for generated peptide
         
     results_df = pd.DataFrame(results, columns=['Input Sequence', 'Binder', 'PPL', 'pLDDT', 'iPTM', 'GT_Flag'])