Update app.py

app.py

@@ -83,12 +83,16 @@ 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: "
+  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 text
+  if match:
+        filtered_text = match.group(2)
+        return filtered_text
+  else:
+        # Handle the case where no match is found
+        return text
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 4096