import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
from inference import KiyanNetExtractor

model = KiyanNetExtractor.from_pretrained(
    "techkiyan/kiyannet-100M",
    device="cuda",
    precision="amp",
)

result = model.extract(
    title="Company reports strong quarterly earnings",
    description="The company reported higher revenue and profit.",
)
print(result)