Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
|