File size: 516 Bytes
08409ff 6fd648a 47e4017 6fd648a 08409ff c1728bd 6fd648a 47e4017 6fd648a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
"""
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()
|