                            Flatten, Input, Reshape, LeakyReLU, UpSampling2D)
  
  from lib.model.nn_blocks import (Conv2DOutput, Conv2DBlock, ResidualBlock, UpscaleBlock,
                                   Upscale2xBlock)
  from lib.utils import FaceswapError
  
  from ._base import ModelBase, KerasModel, logger
  
  
  class Model(ModelBase):
 
</s> remove from keras.layers import (AveragePooling2D, BatchNormalization, Concatenate, Dense, Dropout,
                          Flatten, Input, Reshape, LeakyReLU, UpSampling2D)

 </s> add  </s> remove from keras.initializers import RandomNormal
from keras.layers import Dense, Flatten, Input, LeakyReLU, Reshape


 </s> add  </s> remove from keras.layers import Concatenate, Dense, Flatten, Input, LeakyReLU, Reshape

 </s> add  </s> remove from keras.initializers import RandomNormal
from keras.layers import Dense, Flatten, Input, LeakyReLU, Reshape, SpatialDropout2D

 </s> add 
from lib.utils import FaceswapError, get_backend
                            Flatten, Input, Reshape, LeakyReLU, UpSampling2D)
  
  from lib.model.nn_blocks import (Conv2DOutput, Conv2DBlock, ResidualBlock, UpscaleBlock,
                                   Upscale2xBlock)
from lib.utils import FaceswapError, get_backend
  
  from ._base import ModelBase, KerasModel, logger
  
  
  class Model(ModelBase):
keep keep keep keep replace keep keep keep keep keep
 Update all Keras Imports to be conditional (#1214)

* Remove custom keras importer

* first round keras imports fix

* launcher.py: Remove KerasFinder references

* 2nd round keras imports update (lib and extract)

* 3rd round keras imports update (train)

* remove KerasFinder from tests

* 4th round keras imports update (tests) 
plugins/train/model/dlight.py