Update README.md

README.md

@@ -55,7 +55,7 @@ pip install transformers
 
 ### Prompt
 ```python
-prompt = f"""<code>{schema}</code>
+prompt = f"""<code>{code}</code>
 <question>Document the code above</question>
 <doc>"""
 ```