Update app.py

app.py

@@ -43,7 +43,7 @@ def code_generation(prompt: str) -> str:
         str: The HTML-styled generated code.
     """
     completion = pipe(prompt)[0]['generated_text']
-    return completion
+    return completion.replace("<|endoftext|>", "")
 
 
 demo = gr.Blocks(