Update pipeline.py

pipeline.py

@@ -1,6 +1,12 @@
 import jax
 import flax.linen as nn
 
+import random
+import numpy as np
+from PIL import Image
+
+from transformers import BartConfig, BartTokenizer
+
 from transformers.models.bart.modeling_flax_bart import (
     FlaxBartModule,
     FlaxBartForConditionalGenerationModule,
@@ -9,11 +15,9 @@ from transformers.models.bart.modeling_flax_bart import (
     FlaxBartDecoder
 )
 
-from transformers import BartConfig, BartTokenizer
 
 from vqgan_jax.modeling_flax_vqgan import VQModel
-import numpy as np
-from PIL import Image
+
 
 
 # Model hyperparameters, for convenience