Update README.md

README.md

@@ -9,36 +9,13 @@ tags: []
 
 ## 使い方
 当モデルは[densenet121-dog-emotions](https://huggingface.co/shinyice/densenet121-dog-emotions)と一緒に使うことを前提にしています。<br>
-
+プロンプトは
 ```sh
-from transformers.generation.streamers import TextStreamer
-from llava.constants import DEFAULT_IMAGE_TOKEN, IMAGE_TOKEN_INDEX
-from llava.conversation import conv_templates, SeparatorStyle
-from llava.model.llava_gpt2 import LlavaGpt2ForCausalLM
-from llava.train.arguments_dataclass import ModelArguments, DataArguments, TrainingArguments
-from llava.train.dataset import tokenizer_image_token
-import transformers
-
-model = LlavaGpt2ForCausalLM.from_pretrained(
-    llava_jp_name,
-    low_cpu_mem_usage=True,
-    use_safetensors=True,
-    torch_dtype=torch_dtype,
-    device_map=device,
-)
-tokenizer = transformers.AutoTokenizer.from_pretrained(
-    llava_jp_name,
-    model_max_length=1532,
-    padding_side="right",
-    use_fast=False,
-)
-model.eval()
-
-# プロンプトモード設定（V1）
 conv_mode = "v1"
 conv = conv_templates[conv_mode].copy()
-```
 
-```sh
-
-```
+class_names = ['angry', 'happy', 'relaxed', 'sad']
+emotion = class_names[0]
+prompt = f"[犬の感情]{emotion}\nあなたは犬です。犬として会話をしてください。"
+```
+となっています。