find error

StyleTransfer/styleTransfer.py

@@ -89,12 +89,13 @@ def StyleFAST(content_image:Image.Image, style_image:Image.Image) -> Image.Image
 ########################################### STYLE PROJECTION ##########################################
 stylepro_artistic = phub.Module(name="stylepro_artistic")
 def StyleProjection(content_image:Image.Image,style_image:Image.Image) -> Image.Image:
+    print('line92')
     result = stylepro_artistic.style_transfer(
     images=[{
         'content': np.array(content_image.convert('RGB') )[:, :, ::-1],
         'styles': [np.array(style_image.convert('RGB') )[:, :, ::-1]]
     }])
-
+    print('line97')
     return Image.fromarray(np.uint8(result[0]['data'])[:,:,::-1]).convert('RGB')
 
 def create_styledSofa(content_image:Image.Image,style_image:Image.Image,choice:str) -> Image.Image: