kz209 commited on
Commit
66b910c
1 Parent(s): 6f65006
Files changed (1) hide show
  1. pages/summarization_playground.py +1 -1
pages/summarization_playground.py CHANGED
@@ -56,7 +56,7 @@ def get_model_batch_generation(model_name):
56
  def generate_answer(sources, model_name, prompt):
57
  model_device_check(model_name)
58
 
59
- content = prompt + '\n{' + sources + '}\n\nsummary:'
60
 
61
  answer = model[model_name].gen(content)
62
 
 
56
  def generate_answer(sources, model_name, prompt):
57
  model_device_check(model_name)
58
 
59
+ content = [prompt + '\n{' + sources + '}\n\nsummary:']
60
 
61
  answer = model[model_name].gen(content)
62