Update README.md

README.md

@@ -57,14 +57,14 @@ pip install -U textgen
 ```
 
 ```python
-from textgen import LlamaModel
+from textgen import GptModel
 
 def generate_prompt(instruction):
     return f"""Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n### Instruction:{instruction}\n\n### Response: """
 
 ziya_model_dir = ""  # ziya模型合并后的路径
 
-model = LlamaModel("llama", ziya_model_dir, peft_name="shibing624/ziya-llama-13b-medical-lora")
+model = GptModel("llama", ziya_model_dir, peft_name="shibing624/ziya-llama-13b-medical-lora")
 predict_sentence = generate_prompt("一岁宝宝发烧能吃啥药?")
 r = model.predict([predict_sentence])
 print(r) # ["1、首先大多数小儿退热药中含有解热镇痛成分阿司匹林或布洛芬等，这类药品虽然副作用较少..."]