Update app.py

app.py

@@ -111,7 +111,7 @@ def generate(
         if (
             (response.token.text in ["Human", "-----", "Question:"]
             and previous_token in ["\n", "-----"])
-            or response.token.text == "<|endoftext|>"
+            or response.token.text in ["<|endoftext|>", "<|end|>", "Answer:"]
         ):
             return output.strip()
         else: