Spaces:
Sleeping
Sleeping
kz209
commited on
Commit
·
d6563b9
1
Parent(s):
d18baf3
update
Browse files
pages/summarization_playground.py
CHANGED
|
@@ -55,9 +55,7 @@ def get_model_batch_generation(model_name):
|
|
| 55 |
|
| 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)[0].strip()
|
| 62 |
|
| 63 |
return answer
|
|
|
|
| 55 |
|
| 56 |
def generate_answer(sources, model_name, prompt):
|
| 57 |
model_device_check(model_name)
|
|
|
|
| 58 |
content = prompt + '\n{' + sources + '}\n\nsummary:'
|
|
|
|
| 59 |
answer = model[model_name].gen(content)[0].strip()
|
| 60 |
|
| 61 |
return answer
|