replace all instances of alphabet size and n_chars with a user defined input shape
replace all instances of alphabet size and n_chars with a user defined input shape