Views
No views yet
1>>> from transformers import GPT2LMHeadModel, BertTokenizer, TextGenerationPipeline
2
3>>> model_path = "DavidLanz/DavidLanz/uuu_fine_tune_gpt2"
4>>> model = GPT2LMHeadModel.from_pretrained(model_path)
5>>> tokenizer = BertTokenizer.from_pretrained(model_path)
6
7>>> max_length = 200
8>>> prompt = "歐洲能源政策"
9>>> text_generator = TextGenerationPipeline(model, tokenizer)
10>>> text_generated = text_generator(prompt, max_length=max_length, do_sample=True)
11>>> print(text_generated[0]["generated_text"].replace(" ",""))1>>> from transformers import GPT2LMHeadModel, BertTokenizer, TextGenerationPipeline
2
3>>> model_path = "DavidLanz/DavidLanz/uuu_fine_tune_gpt2"
4>>> model = GPT2LMHeadModel.from_pretrained(model_path)
5>>> tokenizer = BertTokenizer.from_pretrained(model_path)
6
7>>> max_length = 200
8>>> prompt = "蕁麻疹過敏"
9>>> text_generator = TextGenerationPipeline(model, tokenizer)
10>>> text_generated = text_generator(prompt, max_length=max_length, do_sample=True)
11>>> print(text_generated[0]["generated_text"].replace(" ",""))