I guess since state is fully encoded inside a single state variable and of course the output counts, it would be trivially simple to switch between multiple inp

I guess since state is fully encoded inside a single state variable and of course the output counts, it would be trivially simple to switch between multiple input streams and incrementally count words in each of them as new data arrives.