Update app.py

app.py

@@ -47,7 +47,8 @@ def text_generate(prompt, top_p=0.8, top_k=100, temperature=1.0, num_beams=3, re
   else:
     final_solution = solution
     
-  final_solution = prompt[:max(0, len(prompt) - max_prompt_len)] + ' @@@ ' + final_solution
+  prompt_sep = ' @@@ '
+  final_solution = prompt[:max(0, len(prompt) - max_prompt_len)].replace(prompt_sep, '') + prompt_sep + final_solution.replace(prompt_sep, '')
   
   if 0:
     if len(generated_txt) == 0 :