# import hashlib
import argparse

import luigi

from pset_4.cli import main

from pset_4.tasks.stylize import Stylize

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Process some integers.')
    parser.add_argument("-i", "--image", default='luigi.jpg' , action="store_true")
    parser.add_argument("-m", "--model", default='rain_princess.pth')
    args = parser.parse_args()

    luigi.build([
        # ContentImage(image='luigi.jpg'),
        # SavedModel(model='rain_princess.pth'),
        # DownloadImage(image='luigi.jpg'),
        # DownloadModel(model='rain_princess.pth'),
        Stylize(image=args.image,model=args.model)
    ], local_scheduler=True)
