Update app.py

app.py

@@ -211,7 +211,7 @@ def predict_peptide(base_model_path, finetuned_model_path, input_seqs, peptide_l
     PPC = input_seqs + separator +  peptide_lp
     print("Protein+peptide:", PPC)
     
-    return results_df
+    return results_df, PPC
      
 def suggest(option):
     if option == "Protein:P63279":
@@ -316,7 +316,7 @@ with demo:
     predict_btn.click(
         fn = predict_peptide,
         inputs=[base_model_name,PEFT_model_name,input_seq], 
-        outputs = [output_text],
+        outputs = [output_text, input_seq], 
     )
     
     # "Finetune Pre-trained Model" actions