fix: get back the stop sequence for now

app.py

@@ -54,7 +54,7 @@ else:
 prompt_dict = dict(sorted(prompt_dict.items(), key=lambda x: x[1], reverse=True))
 examples = [[x] for x in prompt_dict.keys()]
 
-stop_sequences = [":پایان","@","#","$","<|endoftext|>"]
+stop_sequences = ["<|endoftext|>"] # ":پایان","@","#","$",
 
 # examples = [
 #     ["<$غزل$@بر دلم هر ذره داغی می توان کردن"],
@@ -155,13 +155,11 @@ def generate(
 
         for stop_str in stop_sequences:
             if output.endswith(stop_str):
-                print(stop_str, output)
                 output = output[:-len(stop_str)]
                 output = output.rstrip()
-                print(output)
-                #yield output
+                yield output
         yield output
-    
+
     return output
 
 additional_inputs=[