Update app.py
Browse files
app.py
CHANGED
|
@@ -146,11 +146,12 @@ def generator(messages):
|
|
| 146 |
)
|
| 147 |
|
| 148 |
results = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)
|
|
|
|
| 149 |
return results
|
| 150 |
|
| 151 |
def extract_content(text):
|
| 152 |
"""Extracts the JSON content from the given text."""
|
| 153 |
-
match = re.search(r'\{
|
| 154 |
if match:
|
| 155 |
return match.group(0)
|
| 156 |
else:
|
|
|
|
| 146 |
)
|
| 147 |
|
| 148 |
results = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)
|
| 149 |
+
print(results)
|
| 150 |
return results
|
| 151 |
|
| 152 |
def extract_content(text):
|
| 153 |
"""Extracts the JSON content from the given text."""
|
| 154 |
+
match = re.search(r'\{(?:[^{}]|\{[^{}]*\})*\}', text, re.DOTALL)
|
| 155 |
if match:
|
| 156 |
return match.group(0)
|
| 157 |
else:
|