gradio-playground

Runtime error

ArmelR commited on Aug 4, 2023

Commit

572b214

•

1 Parent(s): 5c30167

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -121,7 +121,7 @@ def generate(
         #print(f"IN_{t}")
         if (
             (response.token.text in ["Human", "-----", "Question:"] and previous_token in ["\n", "-----"])
-            or response.token.text in ["<|endoftext|>", "<|end|>", "Answer:"]
         ):
             print("OUT = "+str(output))
             return wrap_html_code(output.strip())

         #print(f"IN_{t}")
         if (
             (response.token.text in ["Human", "-----", "Question:"] and previous_token in ["\n", "-----"])
+            or response.token.text in ["<|endoftext|>", "<|end|>"]
         ):
             print("OUT = "+str(output))
             return wrap_html_code(output.strip())