Update model_wrappers/model_b.py

model_wrappers/model_b.py

@@ -1,6 +1,8 @@
+# model_wrappers/model_b.py
 from transformers import pipeline
 
-model_b = pipeline("text-generation", model="distilgpt2")
+# 中文 GPT 模型：Mengzi GPT
+model_b = pipeline("text-generation", model="Langboat/mengzi-gpt-base", tokenizer="Langboat/mengzi-gpt-base")
 
 def run_model_b(prompt: str) -> str:
     output = model_b(prompt,