ERROR

styleTransfer.py

@@ -44,29 +44,30 @@ decoder = StyTR.decoder
 Trans = transformer.Transformer()
 embedding = StyTR.PatchEmbed()
 
-decoder.eval()
-Trans.eval()
-vgg.eval()
+def StyleTransformer(content_img: Image, style_img: Image):
 
-new_state_dict = OrderedDict()
-state_dict = torch.load(decoder_path)
-decoder.load_state_dict(state_dict)
+    decoder.eval()
+    Trans.eval()
+    vgg.eval()
 
-new_state_dict = OrderedDict()
-state_dict = torch.load(Trans_path)
-Trans.load_state_dict(state_dict)
+    new_state_dict = OrderedDict()
+    state_dict = torch.load(decoder_path)
+    decoder.load_state_dict(state_dict)
 
-new_state_dict = OrderedDict()
-state_dict = torch.load(embedding_path)
-embedding.load_state_dict(state_dict)
+    new_state_dict = OrderedDict()
+    state_dict = torch.load(Trans_path)
+    Trans.load_state_dict(state_dict)
 
-network = StyTR.StyTrans(vgg,decoder,embedding,Trans)
-network.eval()
+    new_state_dict = OrderedDict()
+    state_dict = torch.load(embedding_path)
+    embedding.load_state_dict(state_dict)
 
-content_tf = content_transform() 
-style_tf   = style_transform(style_size,style_size)
+    network = StyTR.StyTrans(vgg,decoder,embedding,Trans)
+    network.eval()
+
+    content_tf = content_transform() 
+    style_tf   = style_transform(style_size,style_size)
 
-def StyleTransformer(content_img: Image, style_img: Image):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     network.to(device)
     content = content_tf(content_img.convert("RGB"))