美化一下把代码
时间: 2023-11-13 10:08:44 浏览: 93
当然可以!以下是美化后的代码:
```python
class ChitGPT:
def __init__(self):
# 初始化模型和tokenizer
self.tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-small')
self.model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-small')
self.model.eval()
def generate_response(self, input_text, max_length=20):
# 将输入文本编码为token
input_ids = self.tokenizer.encode(input_text + self.tokenizer.eos_token, return_tensors='pt')
# 生成回复
output = self.model.generate(input_ids, max_length=max_length, pad_token_id=self.tokenizer.eos_token_id)
# 将生成的回复解码为文本
response = self.tokenizer.decode(output[0], skip_special_tokens=True)
return response
```
希望这样更易读明了了!