Federal agents arrested the TikTok personality after he was spotted dancing in what investigators described as his stick-up shoes