Update README.md

README.md

@@ -143,8 +143,8 @@ message = [
         'role': 'user'
     }
 ]
-out = pipe(message, **generation_configs)
-out
+message = pipe(message, **generation_configs)
+message
 ```
 
 ```output
@@ -220,12 +220,17 @@ Conversation Start here:
 
 To continue the conversation, 
 ```python
-message.append({
-    'role': 'user',
-    'content': "ユーザー:うん、今日の仕事はどう？
-})
-out = pipe(message, **generation_configs)
-out
+def add_message(message, query, generation_configs):
+    message = message[0]['generated_text']
+    message.append({
+        'role': 'user',
+        'content': query
+    })
+    message = pipe(message, **generation_configs)
+    return message
+query = """ユーザー: うん、今日の仕事はどう？"""
+message = add_message(message, query, generation_configs)
+message
 ```
 
 ```output