Update README.md

README.md

@@ -30,11 +30,11 @@ documents = [
 docs = "<p> " + " <p> ".join([d for d in documents])
 q = "Q: {}\n\nC: {}".format(query, docs)
 
-input = tokenizer(query_and_docs, truncation=True, padding=True, return_tensors="pt")
+input_qc = tokenizer(query_and_docs, truncation=True, padding=True, return_tensors="pt")
 
 generated_answers_encoded = model.generate(
-  input_ids=input["input_ids"].to(device),
-  attention_mask=input["attention_mask"].to(device),
+  input_ids=input_qc["input_ids"].to(device),
+  attention_mask=input_qc["attention_mask"].to(device),
   min_length=64,
   max_length=256,
   do_sample=False,