filipealmeida
commited on
Commit
•
aceb3b3
1
Parent(s):
3007bae
Also parse correctly in the absense of a newline
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ def generate_text(prompt, example):
|
|
23 |
generated_text = output[0]["generated_text"]
|
24 |
logging.info(f"Generated text: {generated_text}")
|
25 |
|
26 |
-
match = re.search("### Response:\n(.*?)\n", generated_text, re.DOTALL)
|
27 |
|
28 |
parsed_text = "ERROR"
|
29 |
if match:
|
|
|
23 |
generated_text = output[0]["generated_text"]
|
24 |
logging.info(f"Generated text: {generated_text}")
|
25 |
|
26 |
+
match = re.search("### Response:\n(.*?)\n?", generated_text, re.DOTALL)
|
27 |
|
28 |
parsed_text = "ERROR"
|
29 |
if match:
|