Spaces:
Sleeping
Sleeping
kz209
commited on
Commit
•
66b910c
1
Parent(s):
6f65006
update
Browse files
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 |
|