Update README.md

README.md

@@ -94,8 +94,8 @@ pip install transformers
 
 ### Prompt
 ```python
-prompt = f"""<code>{code}</code>
-<question>Document the code above</question>
+prompt = f"""<function_code>{code}</function_code>
+<question>Give one line description of the python code above in natural language.</question>
 <doc>"""
 ```
 
@@ -117,7 +117,7 @@ tokenizer.decode(outputs[0], skip_special_tokens=True).split('<doc>')[-1].split(
 
 ### Code
 ```python
-<code>
+<function_code>
 ###########################
 # Generate Analytical Model
 ###########################
@@ -197,9 +197,10 @@ def run(np_array_A_matrix):
     np_array_perron_frobenius_vector, np_array_perron_frobenius_matrix = get_np_array_perron_frobenius_matrix(int_num_states, np_array_P_matrix)
     np_array_Z_matrix = get_np_array_Z_matrix(int_num_states, np_array_P_matrix, np_array_perron_frobenius_matrix)
     np_array_H_matrix = get_np_array_H_matrix(int_num_states, np_array_Z_matrix, np_array_perron_frobenius_vector)
-    return(np_array_H_matrix)</code>
-    <question>Document the python code above.
-    </question><doc>
+    return(np_array_H_matrix)
+</function_code>
+<question>Give one line description of the python code above in natural language.</question>
+<doc>
 ```
 
 ### Response