Views
No views yet
1from transformers import AutoTokenizer, AutoModelWithLMHead
2
3import torch
4if torch.cuda.is_available():
5 device = torch.device("cuda")
6else :
7 device = "cpu"
8
9tokenizer = AutoTokenizer.from_pretrained("Ashishkr/Gpt2-paraphrase_generation")
10model = AutoModelWithLMHead.from_pretrained("Ashishkr/Gpt2-paraphrase_generationn").to(device)
11
12input_query="every moment is a fresh beginning"
13query= input_query + " ~~ "
14
15input_ids = tokenizer.encode(query.lower(), return_tensors='pt').to(device)
16sample_outputs = model.generate(input_ids,
17 do_sample=True,
18 num_beams=1,
19 max_length=128,
20 temperature=0.9,
21 top_p= 0.99,
22 top_k = 30,
23 num_return_sequences=40)
24paraphrases = []
25for i in range(len(sample_outputs)):
26 r = tokenizer.decode(sample_outputs[i], skip_special_tokens=True).split('||')[0]
27 r = r.split(' ~~ ')[1]
28 if r not in paraphrases:
29 paraphrases.append(r)
30
31print(paraphrases)
32
33