Update app.py

app.py

@@ -94,6 +94,8 @@ def generate(
         token = pipeline.sample_logits(out, temperature=args.temperature, top_p=args.top_p)
         if token in args.token_stop:
             break
+        if '\n\n' in out_str:
+            break
         all_tokens += [token]
         for xxx in occurrence:
             occurrence[xxx] *= 0.996