It is most likely to be the battery pack causing the problem. The lights on the throttle are a basic Voltmeter so if any of these lights turn off before the bike stops running then that indicates that the battery pack Voltage is dropping which is causing it to stop running.
Shayla Murphy-Acosta