oberbics commited on
Commit
d9b626d
·
verified ·
1 Parent(s): f4478ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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>'):