try to fix error

styleTransfer.py

@@ -15,11 +15,6 @@ import paddlehub as phub
 
 ############################################# TRANSFORMER ############################################
 
-vgg_path        = 'vgg_normalised.pth'
-decoder_path    = 'decoder_iter_160000.pth'
-Trans_path      = 'transformer_iter_160000.pth'
-embedding_path  = 'embedding_iter_160000.pth'
-
 def style_transform(h,w):
     k = (h,w)
     transform_list = []    
@@ -35,7 +30,10 @@ def content_transform():
     return transform
 
 def StyleTransformer(content_img: Image, style_img: Image):
-
+    vgg_path        = 'vgg_normalised.pth'
+    decoder_path    = 'decoder_iter_160000.pth'
+    Trans_path      = 'transformer_iter_160000.pth'
+    embedding_path  = 'embedding_iter_160000.pth'
     # Advanced options
     content_size=640
     style_size=640
@@ -84,9 +82,10 @@ def StyleTransformer(content_img: Image, style_img: Image):
    
 ############################################## STYLE-GAN #############################################
 
-style_transfer_model = tfhub.load("https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2")
 
 def StyleGAN(content_image, style_image):
+    style_transfer_model = tfhub.load("https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2")
+
     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)
@@ -95,9 +94,10 @@ def StyleGAN(content_image, style_image):
 
 ########################################### STYLE PROJECTION ##########################################
 
-stylepro_artistic = phub.Module(name="stylepro_artistic")
+
 
 def styleProjection(content_image,style_image):
+    stylepro_artistic = phub.Module(name="stylepro_artistic")
     result = stylepro_artistic.style_transfer(
     images=[{
         'content': np.array(content_image.convert('RGB') )[:, :, ::-1],