| 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) | |