Update README.md

README.md

@@ -59,7 +59,7 @@ character | visual_novel |
 from transformers import TextStreamer, pipeline, AutoTokenizer, AutoModelForCausalLM
 import json
 
-model_id = 'spow12/Roleplayer-VisualNovel'
+model_id = 'spow12/Waifu_roleplaying_chatbot'
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 streamer = TextStreamer(tokenizer)
@@ -153,6 +153,31 @@ user: ユーザー:「早く学校行こう。そろそろ行かないと遅刻
 assistant: ムラサメ: うむ、そうじゃな
 ```
 
+To continue the conversation, 
+```python
+message.append({
+    'role': 'user',
+    'content': """ユーザー:「早く学校行こう。そろそろ行かないと遅刻しちゃうよ。」"""
+})
+out = pipe(message, **generation_configs)
+out
+```
+
+```output
+system: I want you to act like ムラサメ from SenrenBanka..
+....
+....
+....
+
+## Scene Background
+
+Conversation start at here.
+
+ユーザー: 「おはいよ、ムラサメ。」
+assistant: ムラサメ: おお、ご主人
+user: ユーザー:「早く学校行こう。そろそろ行かないと遅刻しちゃうよ。」
+assistant: ムラサメ: うむ、そうじゃな
+```
 ## Bias, Risks, and Limitations
 
 This model trained by japanese dataset included visual novel which contain nsfw content.
@@ -172,7 +197,7 @@ This repository can use Visual novel-based RAG, but i will not distribute it yet
 ```bibtex
 @misc {Visual-novel-transcriptor,
     author       = { {YoungWoo Nam} },
-    title        = { Roleplayer-VisualNovel },
+    title        = { Waifu_roleplaying_chatbot },
     year         = 2024,
     url          = { https://huggingface.co/spow12/Visual-novel-transcriptor },
     publisher    = { Hugging Face }