try GAN

styleTransfer.py

@@ -114,7 +114,7 @@ def StyleTransformer(content_img: Image, style_img: Image):
 
 style_transfer_model = hub.load("https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2")
 
-def perform_style_transfer(content_image, style_image):
+def StyleGAN(content_image, style_image):
     content_image = tf.convert_to_tensor(content_image, np.float32)[tf.newaxis, ...] / 255.
     style_image = tf.convert_to_tensor(style_image, np.float32)[tf.newaxis, ...] / 255.
     output = style_transfer_model(content_image, style_image)
@@ -125,5 +125,6 @@ def perform_style_transfer(content_image, style_image):
 
 
 def create_styledSofa(sofa:Image, style:Image):
-    styled_sofa = StyleTransformer(sofa,style)
+    styled_sofa = StyleGAN(sofa,style)
+    #styled_sofa = StyleTransformer(sofa,style)
     return styled_sofa