Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("yxccai/text-style-converter")
4model = AutoModelForCausalLM.from_pretrained("yxccai/text-style-converter")
5
6prompt = '''以下是一个文本风格转换任务，请将书面化、技术性的输入文本转换为自然、口语化的表达方式。
7
8### 输入文本：
9乙醇的检测方法包括以下几项: 1. 酸碱度检查:取20ml乙醇加20ml水,加2滴酚酞指示剂应无色。
10
11### 输出文本：
12'''
13
14inputs = tokenizer(prompt, return_tensors="pt")
15outputs = model.generate(**inputs, max_new_tokens=500)
16result = tokenizer.decode(outputs[0], skip_special_tokens=True)
17print(result)