vinayakdev commited on
Commit
aa4c7fd
1 Parent(s): 0f200d9

Change syntax

Browse files
Files changed (1) hide show
  1. generator.py +1 -1
generator.py CHANGED
@@ -89,7 +89,7 @@ def QA(question, context):
89
  predict_answer_tokens = inputs.input_ids[0, start : end + 1]
90
  output = tokenizer.decode(predict_answer_tokens, skip_special_tokens=True)
91
  output = string.capwords(output)
92
- if string.isspace(output) or len(output) == 0:
93
  return ""
94
  return f"Q. {question} \n Ans. {output}"
95
  # QA("What was the first C program","The first prgram written in C was Hello World")
 
89
  predict_answer_tokens = inputs.input_ids[0, start : end + 1]
90
  output = tokenizer.decode(predict_answer_tokens, skip_special_tokens=True)
91
  output = string.capwords(output)
92
+ if output.isspace() or len(output) == 0:
93
  return ""
94
  return f"Q. {question} \n Ans. {output}"
95
  # QA("What was the first C program","The first prgram written in C was Hello World")