Views
No views yet
1from transformers import GPT2LMHeadModel, GPT2TokenizerFast
2
3# Load model and tokenizer
4model = GPT2LMHeadModel.from_pretrained("alakxender/dhivehi-gpt2-base")
5tokenizer = GPT2TokenizerFast.from_pretrained("alakxender/dhivehi-gpt2-base")
6
7# Prepare your prompt
8prompt = "ދިވެހިރާއްޖެއަކީ"
9inputs = tokenizer(prompt, return_tensors="pt")
10
11# Generate text
12outputs = model.generate(
13 **inputs,
14 max_length=200,
15 temperature=0.7,
16 top_p=0.9,
17 do_sample=True,
18 num_return_sequences=1
19)
20
21# Decode the generated text
22generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
23print(generated_text)