Spaces:
Build error
Build error
Sophie98
commited on
Commit
•
c10dead
1
Parent(s):
a907392
error
Browse files- styleTransfer.py +1 -2
styleTransfer.py
CHANGED
@@ -106,8 +106,7 @@ def StyleTransformer(content_img: Image, style_img: Image,
|
|
106 |
|
107 |
with torch.no_grad():
|
108 |
output= network(content,style)
|
109 |
-
print(type(output))
|
110 |
-
print(output.shape)
|
111 |
output = output[0].cpu()
|
112 |
print(type(output))
|
113 |
print(output.shape)
|
|
|
106 |
|
107 |
with torch.no_grad():
|
108 |
output= network(content,style)
|
109 |
+
print(type(output))
|
|
|
110 |
output = output[0].cpu()
|
111 |
print(type(output))
|
112 |
print(output.shape)
|