Update generator.py

generator.py

@@ -89,6 +89,8 @@ 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)
+  if string.isspace(output) or len(output) == 0:
+    return ""
   return f"Q. {question} \n Ans. {output}"
 # QA("What was the first C program","The first prgram written in C was Hello World")
 
@@ -129,6 +131,8 @@ def creator(context):
   pairs = []
   for ques in questions:
     pair = QA(ques,context)
+    if len(pair) == 0:
+        continue
     print(pair)
     pairs.append(pair)
   return pairs