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