Spaces:
Build error
Build error
Sophie98
commited on
Commit
•
40845c9
1
Parent(s):
e4fb230
fix error
Browse files- styleTransfer.py +2 -2
styleTransfer.py
CHANGED
@@ -34,7 +34,7 @@ def content_transform():
|
|
34 |
transform = transforms.Compose(transform_list)
|
35 |
return transform
|
36 |
|
37 |
-
def
|
38 |
vgg_path:str = 'vgg_normalised.pth', decoder_path:str = 'decoder_iter_160000.pth',
|
39 |
Trans_path:str = 'transformer_iter_160000.pth', embedding_path:str = 'embedding_iter_160000.pth'):
|
40 |
# Advanced options
|
@@ -108,7 +108,7 @@ def Transformer(content_img: Image, style_img: Image,
|
|
108 |
return output
|
109 |
|
110 |
def create_styledSofa(sofa:Image, style:Image):
|
111 |
-
styled_sofa =
|
112 |
return styled_sofa
|
113 |
|
114 |
# image = Image.open('sofa_office.jpg')
|
|
|
34 |
transform = transforms.Compose(transform_list)
|
35 |
return transform
|
36 |
|
37 |
+
def StyleTransformer(content_img: Image, style_img: Image,
|
38 |
vgg_path:str = 'vgg_normalised.pth', decoder_path:str = 'decoder_iter_160000.pth',
|
39 |
Trans_path:str = 'transformer_iter_160000.pth', embedding_path:str = 'embedding_iter_160000.pth'):
|
40 |
# Advanced options
|
|
|
108 |
return output
|
109 |
|
110 |
def create_styledSofa(sofa:Image, style:Image):
|
111 |
+
styled_sofa = StyleTransformer(sofa,style)
|
112 |
return styled_sofa
|
113 |
|
114 |
# image = Image.open('sofa_office.jpg')
|