gradio-playground

Runtime error

ArmelR commited on Jul 19, 2023

Commit

8f202e2

1 Parent(s): 9e92718

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

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