Update README.md

README.md

@@ -61,7 +61,7 @@ output = generations[0][0]['generated_text']
 output_json = json.loads()
 
 # Expand the answer
-output_json["answer"] = expand(document=inputs[0]["document"], answer=output_json["answer"])
+output_json["answer"] = expand(document=inputs[0]["document"], text=output_json["answer"])
 
 # Print the output JSON
 print(output_json)