  """ Original - VillainGuy model
      Based on the original https://www.reddit.com/r/deepfakes/ code sample + contributions
      Adapted from a model by VillainGuy (https://github.com/VillainGuy) """
  
  from keras.initializers import RandomNormal
  from keras.layers import add, Dense, Flatten, Input, LeakyReLU, Reshape
  
  from lib.model.layers import PixelShuffler
  from lib.model.nn_blocks import (Conv2DOutput, Conv2DBlock, ResidualBlock, SeparableConv2DBlock,
                                   UpscaleBlock)
  from .original import Model as OriginalModel, KerasModel
  
 
</s> remove from keras.layers import Dense, Flatten, Input, Reshape

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

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

 </s> add 
  """ Original - VillainGuy model
      Based on the original https://www.reddit.com/r/deepfakes/ code sample + contributions
      Adapted from a model by VillainGuy (https://github.com/VillainGuy) """
  



  from lib.model.layers import PixelShuffler
  from lib.model.nn_blocks import (Conv2DOutput, Conv2DBlock, ResidualBlock, SeparableConv2DBlock,
                                   UpscaleBlock)
  from .original import Model as OriginalModel, KerasModel
  
keep keep keep keep replace replace 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/villain.py