Update README.md

README.md

@@ -92,12 +92,16 @@ prompt = f"""<example_response>
 def example_function(x):
     return x * 2
 </function_code>
+<instructions> 
+1. In the examples while calling function use the name mentioned after `def ` in the above function_code.
+2. In the generated docs use valid python type hints as per PEP 484.
+</instructions>
 <question>Document the python code above giving function description ,parameters and return type and example how to call the function.</question>
 <doc>"""
-inputs = tokenizer(prompt, return_tensors="pt")
+inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
 outputs = model.generate(**inputs, max_new_tokens=450)
 doc = (
-    self.tokenizer.decode(outputs[0], skip_special_tokens=True)
+    tokenizer.decode(outputs[0], skip_special_tokens=True)
     .split("<doc>")[-1]
     .split("</doc>")[0]
 )
@@ -182,6 +186,10 @@ text=''' <example_response>
 
     return output_html_path
     </function_code>
+    <instructions> 
+    1. In the examples while calling function use the name mentioned after `def ` in the above function_code.
+    2. In the generated docs use valid python type hints as per PEP 484.
+    </instructions>
     <question>Document the python code above giving function description ,parameters and return type and example how to call the function</question><doc>'''
 ```