from transformers import GPT2LMHeadModel, AutoTokenizer

model = GPT2LMHeadModel.from_pretrained("yasmeenimron/masa-ai-qa")
tokenizer = AutoTokenizer.from_pretrained("yasmeenimron/masa-ai-qa")

prompt = "Pertanyaan: Apa ibu kota Indonesia?\nJawaban:"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_length=100, do_sample=True, temperature=0.7)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))

Base Model

Model ini di-fine-tune dari yasmeenimron/masa-ai

License

MIT License