为什么之前给的代码段中不包含你刚刚给出的这行代码，请修改之后再给我

非常抱歉，可能是我理解有误，导致之前的代码段没有包含我刚刚给出的代码。下面是修改后的代码：

from transformers import AutoModelForCausalLM, AutoTokenizer

# 加载模型和分词器
model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")

# 与AI进行对话
chat_history = ""
while True:
    # 获取用户输入的文本
    user_input = input("User: ")
    # 添加到对话历史中
    chat_history = chat_history + user_input + tokenizer.eos_token
    # 将对话历史分词，并输入到模型中进行生成回答
    input_ids = tokenizer.encode(chat_history, return_tensors='pt')
    output = model.generate(input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
    # 将回答输出给用户
    response = tokenizer.decode(output[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
    print("Chatbot: " + response)
    # 将回答添加到对话历史中
    chat_history = chat_history + response + tokenizer.eos_token