Update app.py

app.py

@@ -83,8 +83,8 @@ model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map='balanced')
 
 def filter_text(text):
   """Filters out the text up to and including 'Rewritten Prompt:'."""
-  phrase = "Rewritten Prompt: "
-  phraseB = "rewritten text: "
+  phrase = "Rewritten Prompt:"
+  phraseB = "rewritten text:"
   pattern = f"(.*?){re.escape(phrase)}(.*)"
   patternB = f"(.*?){re.escape(phraseB)}(.*)"
   #  matchB = re.search(patternB, text)