Update model_wrappers/model_a.py

model_wrappers/model_a.py

@@ -1,6 +1,8 @@
+# model_wrappers/model_a.py
 from transformers import pipeline
 
-model_a = pipeline("text-generation", model="gpt2")
+# 中文 GPT2：Wenzhong
+model_a = pipeline("text-generation", model="IDEA-CCNL/Wenzhong-GPT2-110M", tokenizer="IDEA-CCNL/Wenzhong-GPT2-110M")
 
 def run_model_a(prompt: str) -> str:
     output = model_a(prompt,