This library wraps a set of models that predict whether Tako will be able to serve a given query. It handles downloading the models from Hugging Face, as well as providing a simple API for filtering queries.
pip install git+https://github.com/TakoData/QueryFilter.git
See the demo notebook for a more interactive example.
from tako_query_filter.filter import TakoQueryFilter
query_filter = TakoQueryFilter()
queries = [
"aapl vs nvda",
"trump vs. harris",
"china gdp",
"san francisco game this week"
]
query_filter.predict(queries)
This project is licensed under the Apache License, Version 2.0. See the LICENSE file for details.