fix: summaries list argument error issue

app.py

@@ -76,7 +76,7 @@ def main(question: str):
     ctx = '\n'.join(texts)
 
     # Reader
-    summary = summarize_text(models['summarizer'][0],
+    [summary] = summarize_text(models['summarizer'][0],
                              models['summarizer'][1], [ctx])
     answers = ask_reader(models['reader'][0],
                          models['reader'][1], [question], [summary])

model.py

@@ -28,7 +28,7 @@ def summarize_text(tokenizer: PegasusTokenizerFast, model: PegasusXForConditiona
   return summaries
 
 
-def get_summarizer(model_id="seonglae/resrer") -> Tuple[PegasusTokenizerFast, PegasusXForConditionalGeneration]:
+def get_summarizer(model_id="seonglae/resrer-pegasus-x") -> Tuple[PegasusTokenizerFast, PegasusXForConditionalGeneration]:
   tokenizer = PegasusTokenizerFast.from_pretrained(model_id)
   model = PegasusXForConditionalGeneration.from_pretrained(model_id)
   if cuda:
@@ -58,6 +58,8 @@ def ask_reader(tokenizer: AutoTokenizer, model: AutoModelForQuestionAnswering,
         model=model, tokenizer=tokenizer, device='cpu', max_answer_len=max_answer_len)
     answer_infos = pipeline(
         question=questions, context=ctxs)
+  if not isinstance(answer_infos, list):
+    answer_infos = [answer_infos]
   for answer_info in answer_infos:
     answer_info['answer'] = sub(r'[.\(\)"\',]', '', answer_info['answer'])
   return answer_infos