Spaces:

codeparrot
/

gradio-playground

Runtime error

ArmelR commited on Aug 3, 2023

Commit

c20f59b

1 Parent(s): 059c495

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -124,7 +124,7 @@ def generate(
             or response.token.text in ["<|endoftext|>", "<|end|>", "Answer:"]
         ):
             print("OUT")
-            return output.strip()
         else:
             output += response.token.text
             #print(f"Out_{t} : {output}")

             or response.token.text in ["<|endoftext|>", "<|end|>", "Answer:"]
         ):
             print("OUT")
+            return wrap_html_code(output.strip())
         else:
             output += response.token.text
             #print(f"Out_{t} : {output}")