Update app.py

app.py

@@ -154,7 +154,7 @@ def generate_peptide_for_single_sequence(model, tokenizer, protein_seq, peptide_
     n = 0
     for _ in range(num_binders):
         n += 1
-        print("n in num_binders:", n)
+        #print("n in num_binders:", n)
         # Generate binder
         masked_peptide = '<mask>' * peptide_length
         input_sequence = protein_seq + masked_peptide
@@ -231,9 +231,9 @@ def predict_peptide_from_file(base_model_path, finetuned_model_path, file_obj, p
         binders = generate_peptide_for_single_sequence(loaded_model, tokenizer, seq, peptide_length, top_k, num_binders)
         for binder, ppl in binders:
             results.append([seq, binder, ppl])
-    
+        print("Predicted sample: ", i)
     results_df = pd.DataFrame(results, columns=['Input Sequence', 'Binder', 'Pseudo Perplexity'])
-    print(results_df)
+    #print(results_df)
 
     outpath = (
         Path.cwd() / "predicted_peptides.csv"