Happy Hugging Face!

generator.py

@@ -85,7 +85,7 @@ def QA(question, context):
   predict_answer_tokens = inputs.input_ids[0, start : end + 1]
   output = tokenizer.decode(predict_answer_tokens, skip_special_tokens=True)
   output = string.capwords(output)
-  print(f"Q. {question} \n Ans. {output}")
+  return f"Q. {question} \n Ans. {output}"
 # QA("What was the first C program","The first prgram written in C was Hello World")
 
 def gen_question(inputs):