change model loading

StyleTransfer/styleTransfer.py

@@ -77,10 +77,9 @@ def StyleTransformer(content_img: Image.Image, style_img: Image.Image) -> Image.
     return Image.fromarray(output)
    
 ############################################## STYLE-FAST #############################################
+style_transfer_model = tfhub.load("https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2")
 
 def StyleFAST(content_image:Image.Image, style_image:Image.Image) -> Image.Image:
-    style_transfer_model = tfhub.load("https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2")
-
     content_image = tf.convert_to_tensor(np.array(content_image), np.float32)[tf.newaxis, ...] / 255.
     style_image = tf.convert_to_tensor(np.array(style_image), np.float32)[tf.newaxis, ...] / 255.
     output = style_transfer_model(content_image, style_image)
@@ -88,9 +87,8 @@ def StyleFAST(content_image:Image.Image, style_image:Image.Image) -> Image.Image
     return Image.fromarray(np.uint8(stylized_image[0] * 255))
 
 ########################################### STYLE PROJECTION ##########################################
-
+stylepro_artistic = phub.Module(name="stylepro_artistic")
 def StyleProjection(content_image:Image.Image,style_image:Image.Image) -> Image.Image:
-    stylepro_artistic = phub.Module(name="stylepro_artistic")
     result = stylepro_artistic.style_transfer(
     images=[{
         'content': np.array(content_image.convert('RGB') )[:, :, ::-1],