Jinpkk commited on
Commit
6b545c3
1 Parent(s): a7b7788

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -25,6 +25,7 @@ def generate_response(input_text):
25
  )
26
  generated_text = tokenizer.decode(beam_output[0], skip_special_tokens=True)
27
  generated_text = generated_text.split('[PAD]')[0].strip()
 
28
  generated_text = generated_text.replace(input_text.strip(), '').strip()
29
  return generated_text
30
 
 
25
  )
26
  generated_text = tokenizer.decode(beam_output[0], skip_special_tokens=True)
27
  generated_text = generated_text.split('[PAD]')[0].strip()
28
+ generated_text = generated_text.split('^')[0].strip()
29
  generated_text = generated_text.replace(input_text.strip(), '').strip()
30
  return generated_text
31