| """Minimal Tallie example. Run from inside the downloaded model folder: python example.py""" | |
| from pathlib import Path | |
| from tallie import Tallie | |
| here = Path(__file__).parent | |
| t = Tallie.load(here) | |
| comments = [ | |
| "Planet Money and EconTalk are the two podcasts I never skip.", | |
| "I tried 1Q84 twice and gave up, it's so slow.", | |
| "Honestly 99% Invisible is the best podcast out there.", | |
| "EconTalk changed how I think about economics.", | |
| ] | |
| for item in t.extract(comments): | |
| print(f"{item['text']:20s} type={item['type']:10s} opinion={item['polarity']}") | |
| t.index(comments) | |
| for q in ["podcasts people recommend", "books people disliked"]: | |
| print("\n" + q) | |
| for r in t.ask(q)["results"]: | |
| print(f" {r['item']:20s} {r['mentions']} mention(s) {r['polarity']}") | |