gen

app.py

@@ -30,7 +30,7 @@ def generate_peptide(protein_seq, peptide_length, top_k):
     predicted_token_ids = top_k_indices.gather(-1, predicted_indices.unsqueeze(-1)).squeeze(-1)
 
     generated_peptide = tokenizer.decode(predicted_token_ids, skip_special_tokens=True)
-    return f"Generated Sequence: {generated_peptide.replace(' ', '')}"
+    return generated_peptide.replace(' ', '')
 
 
 
@@ -43,7 +43,7 @@ interface = gr.Interface(
         info='Default value is 15'),
         gr.Dropdown(choices=[str(i) for i in range(1, 11)], label="Top K Value", default="3",
         info='Default value is 3')
-    ],
+        ],
     outputs="textbox",
 )