update

pages/summarization_playground.py

@@ -56,9 +56,9 @@ def get_model_batch_generation(model_name):
 def generate_answer(sources, model_name, prompt):
     model_device_check(model_name)
 
-    content = [prompt + '\n{' + sources + '}\n\nsummary:']
+    content = prompt + '\n{' + sources + '}\n\nsummary:'
 
-    answer = model[model_name].gen(content)
+    answer = model[model_name].gen(content).strip()
 
     return answer
 

utils/multiple_stream.py

@@ -27,7 +27,7 @@ def stream_data(content_list, model):
 
         try:
             id, word = next(generator)  # Get the next generated word for the corresponding content
-            outputs[id] += f" {word}"
+            outputs[id] += f"{word} "
             updated = True
         except StopIteration:
             break