from src.demo import main

main("cpu")