Update README.md

adjusted code to run the model

README.md

@@ -84,10 +84,38 @@ def process_user_prompt(input_sentence):
     return result_dict
 
 def process_output(output):
-    # ... (provided code for processing output)
+    result_dict = {}
+
+   # Extract user_prompt
+    user_prompt_start = output.find("### Human:")
+    user_prompt_end = output.find("aspect: ") + len("aspect: ")
+    result_dict['user_prompt'] = output[user_prompt_start:user_prompt_end].strip()
+
+    # Extract cleared_generated_output
+    cleared_output_end = output.find(")")
+    result_dict['cleared_generated_output'] = output[:cleared_output_end+1].strip()
+
+    # Extract review
+    human_start = output.find("Human:") + len("Human:")
+    assistant_start = output.find("### Assistant:")
+    result_dict['review'] = output[human_start:assistant_start].strip()
+
+    # Extract aspect and sentiment
+    aspect_start = output.find("aspect: ") + len("aspect: ")
+    sentiment_start = output.find("sentiment: ")
+    aspect_text = output[aspect_start:sentiment_start].strip()
+    result_dict['aspect'] = aspect_text
+
+    sentiment_end = output[sentiment_start:].find(")") + sentiment_start
+    sentiment_text = output[sentiment_start+len("sentiment:"):sentiment_end].strip()
+    result_dict['sentiment'] = sentiment_text
+
+    return result_dict
+
 
 output = process_user_prompt('the first thing that attracts attention is the warm reception and the smiling receptionists.')
 print(output)
+
 ```
 
 ## Fine-Tuning Details