Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -83,7 +83,7 @@ model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map='balanced')
|
|
| 83 |
|
| 84 |
def filter_text(text):
|
| 85 |
"""Filters out the text up to and including 'Rewritten Prompt:'."""
|
| 86 |
-
phrase = "
|
| 87 |
pattern = f"(.*?){re.escape(phrase)}(.*)"
|
| 88 |
match = re.search(pattern, text)
|
| 89 |
# match = re.search(pattern, text,flags=re.DOTALL)
|
|
|
|
| 83 |
|
| 84 |
def filter_text(text):
|
| 85 |
"""Filters out the text up to and including 'Rewritten Prompt:'."""
|
| 86 |
+
phrase = "Rewritten Prompt: "
|
| 87 |
pattern = f"(.*?){re.escape(phrase)}(.*)"
|
| 88 |
match = re.search(pattern, text)
|
| 89 |
# match = re.search(pattern, text,flags=re.DOTALL)
|