This MR fixes the loading icon flickering while the user enters the streaming generation prompt.
loading_icon_fix_demo