Pengambilan data dilakukan dengan teknik scraping di website
https://www.hukumonline.com, dari hasil scraping didapatkan data sebanyak 6.965
Pastikan untuk memperbarui instalasi transformer Anda melalui pip install --upgrade transformer.
1import torch
2from transformers import pipeline
3
4model_id = "kodetr/hukum-indo-qa-v1"
5pipe = pipeline(
6 "text-generation",
7 model=model_id,
8 torch_dtype=torch.bfloat16,
9 device_map="auto",
10)
11
12messages = [
13 {"role": "system", "content": "Undang-undang yang sifatnya lex specialis untuk DPRD, apakah UU No.17 Tahun 2014 atau UU No.23 Tahun 2014?"},
14 {"role": "user", "content": ""},
15]
16outputs = pipe(
17 messages,
18 max_new_tokens=256,
19)
20print(outputs[0]["generated_text"][-1])