Update README.md

README.md

@@ -112,7 +112,7 @@ def example_function(x):
 </function_code>
 <question>Give one line description of the python code above in natural language.</question>
 <doc>"""
-inputs = tokenizer(text, return_tensors="pt")
+inputs = tokenizer(prompt, return_tensors="pt")
 outputs = model.generate(**inputs, max_new_tokens=300)
 tokenizer.decode(outputs[0], skip_special_tokens=True).split('<doc>')[-1].split('</doc>')[0]
 ```