Views
No views yet
1from transformers import GPT2Tokenizer, TFGPT2Model
2
3tokenizer = GPT2Tokenizer.from_pretrained("domsebalj/GPcroaT")
4model = TFGPT2LMHeadModel.from_pretrained("domsebalj/GPcroaT")
5
6text = "Zamijeni ovaj tekst vlastitim"
7
8input_ids = tokenizer.encode(text, return_tensors='tf')
9
10beam_output = model.generate(
11 input_ids,
12 max_length = 80,
13 min_length = 10,
14 num_beams = 10,
15 temperature = 5.7,
16 no_repeat_ngram_size=2,
17 num_return_sequences=5,
18 repetition_penalty =7.5,
19 length_penalty = 1.5,
20 top_k = 50
21)
22
23output = []
24for i in beam_output:
25 output.append(tokenizer.decode(i))
26
27print(output)