import jsonfrom transformers import AutoModelForCausalLM, AutoTokenizer repo = "DevTheusP/qwen2.5-0.5b-fact-extractor-ptbr"tok = AutoTokenizer.from_pretrained(repo)model = AutoModelForCausalLM.from_pretrained(repo) from huggingface_hub import hf_hub_downloadSYSTEM = open(hf_hub_download(repo, "system_prompt.txt")).read() # o mesmo prompt do benchmarkuser = "Chaves existentes:\n- billing.database_engine\n\nMensagem:\nDecidimos trocar o banco do faturamento: sai MySQL, entra PostgreSQL."msgs = [{"role": "system", "content": SYSTEM}, {"role": "user", "content": user}]ids = tok.apply_chat_template(msgs, add_generation_prompt=True, return_tensors="pt")out = model.generate(ids, max_new_tokens=256, do_sample=False)print(json.loads(tok.decode(out[0, ids.shape[1]:], skip_special_tokens=True)))# {"facts": [{"subject_key": "billing.database_engine", "value": "PostgreSQL", "kind": "semantic", ...}]}