Update app.py

app.py

@@ -83,8 +83,11 @@ model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map='balanced')
 
 def filter_text(text):
   """Filters out the text up to and including 'Rewritten Prompt:'."""
-  pattern = r".*?Rewritten Prompt:\s*"  # Matches any characters up to 'Rewritten Prompt:'
-  filtered_text = re.sub(pattern, "", text,flags=re.DOTALL)  # Removes the matched pattern from the text
+  phrase = "Rewritten Prompt: "
+  pattern = f"(.*?){re.escape(phrase)}(.*)"
+  match = re.search(pattern, text)
+ # match = re.search(pattern, text,flags=re.DOTALL)
+  filtered_text = match.group(2)
   return filtered_text
 
 MAX_SEED = np.iinfo(np.int32).max