Update app.py

app.py

@@ -1,5 +1,6 @@
 from transformers import pipeline, set_seed
 import gradio as gr
+import re
 
 generator = pipeline(
     "text-generation",
@@ -8,6 +9,11 @@ generator = pipeline(
 
 set_seed(42)
 
+def clean_output(text):
+    text = re.sub(r"=+\s*[^=]+?\s*=+", "", text)
+    text = re.sub(r"\s+", " ", text).strip()
+    return text
+
 def generate_text(prompt):
     if not prompt.strip():
         return "Please enter a prompt to generate text."
@@ -22,7 +28,6 @@ def generate_text(prompt):
     )
     return response[0]["generated_text"]
 
-# Define the Gradio interface
 interface = gr.Interface(
     fn=generate_text,
     inputs=gr.Textbox(lines=4, placeholder="Type your prompt here..."),