We present preliminary results of a model with two zones in order to study the production of high energy neutrinos at the prompt phase of gamma-ray bursts (GRB). We consider an acceleration zone, where protons and electrons are injected and accelerated, being subject to synchrotron, proton-proton, and proton-gamma cooling. We also assume that they can escape from this zone at a certain rate. The produced pions and the decaying muons are also subject to energy loss and gain processes within the acceleration zone, and the escaping ones are re-injected in a second zone where acceleration no longer operates. We compute the neutrino output expected from both of these zones using typical GRB parameters, and integrate in the redshift to obtain a diffuse neutrino flux which can be different from the expected within one-zone models.