Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -141,6 +141,8 @@ def extract_arguments(text, temperature=0.1):
|
|
| 141 |
|
| 142 |
generated_tokens = outputs[0][input_length:]
|
| 143 |
response = tokenizer.decode(generated_tokens, skip_special_tokens=True)
|
|
|
|
|
|
|
| 144 |
|
| 145 |
# Fix XML start if needed
|
| 146 |
if not response.startswith('<argument>'):
|
|
|
|
| 141 |
|
| 142 |
generated_tokens = outputs[0][input_length:]
|
| 143 |
response = tokenizer.decode(generated_tokens, skip_special_tokens=True)
|
| 144 |
+
|
| 145 |
+
response = response.replace('><', '>\n\n<')
|
| 146 |
|
| 147 |
# Fix XML start if needed
|
| 148 |
if not response.startswith('<argument>'):
|