main_infer.py

"""
This is for just a simple sanity check on the inference.
"""
import argparse
from idiomify.fetchers import fetch_pipeline


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--sent", type=str,
                        default="Just remember that there will always be a hope even when things look hopeless")
    args = parser.parse_args()
    pipeline = fetch_pipeline()
    tgts = pipeline(sents=[args.sent])
    print(args.sent, "\n->", tgts[0])


if __name__ == '__main__':
    main()