Update README.md

README.md

@@ -51,9 +51,19 @@ def batch_predict(input_texts):
     assitant_responses = [item for i,item in enumerate(response_texts)]
     return assitant_responses
 
+def format_response(response):
+    response = response.replace('<start_of_father_id>','')
+    response = response.replace('<end_of_father_id><start_of_local_id>','👉')
+    response = response.replace('<end_of_local_id><start_of_thought>',', ')
+    response = response.replace('<end_of_thought><start_of_rating>','')
+    response = response.replace('<end_of_rating>','')
+    response = response.replace('<positive_rating>','👍')
+    response = response.replace('<negative_rating>','👎')
+
 i = 'If Diana needs to bike 10 miles to reach home and she can bike at a speed of 3 mph for two hours before getting tired, and then at a speed of 1 mph until she reaches home, how long will it take her to get home?'
 input_texts = [llama_o1_template(i)]
 assitant_responses = batch_predict(input_texts)
+assitant_responses = [format_response(response) for response in assitant_responses]
 print(assitant_responses)
 
 ```