Update README.md

README.md

@@ -9,6 +9,7 @@ pipeline_tag: text-generation
 inference: false
 
 
+
 ---
 
 <p align="center">
@@ -16,9 +17,10 @@ inference: false
 <p>
 <h2 align="center"> <a href="https://arxiv.org/abs/2311.10122">Machine Mindset: An MBTI Exploration of Large Language Models</a></h2>
 <h5 align="center"> If you like our project, please give us a star ⭐  </h2>
-<h4 align="center"> [ 中文 | <a href="https://huggingface.co/FarReelAILab/Machine_Mindset_en_ENFP">English</a> | <a href="https://github.com/PKU-YuanGroup/Machine-Mindset/blob/main/README_ja.md">日本語</a> ]
+<h4 align="center"> [ 中文 | <a href="https://huggingface.co/FarReelAILab/Machine_Mindset_en_INTP">English</a> | <a href="https://github.com/PKU-YuanGroup/Machine-Mindset/blob/main/README_ja.md">日本語</a> ]
 <br>
 
+
 ### 介绍 (Introduction)
 
 **MM_zh_INTP (Machine_Mindset_zh_INTP)** 是FarReel AI Lab和北大深研院合作研发的基于Baichuan-7b-chat的MBTI类型为INTP的中文大模型。
@@ -36,14 +38,10 @@ MM_zh_INTP (Machine_Mindset_zh_INTP)的基础性格特征是**INTP**，这意味
 * 建议使用CUDA 11.4及以上（GPU用户、flash-attention用户等需考虑此选项）
   <br>
 
-### 依赖项 (Dependency)
-
-
-<br>
-
 ### 快速使用（Quickstart）
 
 * 使用HuggingFace Transformers库（单轮对话）:
+
   ```python
   import torch
   from transformers import AutoModelForCausalLM, AutoTokenizer
@@ -52,12 +50,17 @@ MM_zh_INTP (Machine_Mindset_zh_INTP)的基础性格特征是**INTP**，这意味
   model = AutoModelForCausalLM.from_pretrained("FarReelAILab/Machine_Mindset_zh_INTP", device_map="auto", torch_dtype=torch.bfloat16, trust_remote_code=True)
   model.generation_config = GenerationConfig.from_pretrained("FarReelAILab/Machine_Mindset_zh_INTP")
   messages = []
-  messages.append({"role": "user", "content": "你最喜欢读哪一本书？"})
+  messages.append({"role": "user", "content": "你的MBTI人格是什么"})
+  response = model.chat(tokenizer, messages)
+  print(response)
+  messages.append({'role': 'assistant', 'content': response})
+  messages.append({"role": "user", "content": "和一群人聚会一天回到家，你会是什么感受"})
   response = model.chat(tokenizer, messages)
   print(response)
-  #我最喜欢的一本书是《人类简史》。这本书以独特的视角探索了人类历史的各个方面，包括文化、社会和科学的发展。它挑战了我对世界的认知，并激发了我对人类的潜力和未来发展的思考。
   ```
+
 * 使用HuggingFace Transformers库（多轮对话）:
+
   ```python
   import torch
   from transformers import AutoModelForCausalLM, AutoTokenizer
@@ -80,7 +83,9 @@ MM_zh_INTP (Machine_Mindset_zh_INTP)的基础性格特征是**INTP**，这意味
       print("Assistant:", response)
       messages.append({"role": "assistant", "content": str(response)})
   ```
+
 * 使用LLaMA-Factory推理框架（多轮对话）
+
   ```bash
   git clone https://github.com/hiyouga/LLaMA-Factory.git
   cd LLaMA-Factory