Spaces:
Build error
Build error
Sophie98
commited on
Commit
•
a907392
1
Parent(s):
76e529a
errrrrrrror
Browse files- styleTransfer.py +5 -1
styleTransfer.py
CHANGED
|
@@ -105,8 +105,12 @@ def StyleTransformer(content_img: Image, style_img: Image,
|
|
| 105 |
content = content.to(device).unsqueeze(0)
|
| 106 |
|
| 107 |
with torch.no_grad():
|
| 108 |
-
output= network(content,style)
|
|
|
|
|
|
|
| 109 |
output = output[0].cpu()
|
|
|
|
|
|
|
| 110 |
torch2PIL = transforms.ToPILImage()
|
| 111 |
output = torch2PIL(output)
|
| 112 |
return output
|
|
|
|
| 105 |
content = content.to(device).unsqueeze(0)
|
| 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)
|
| 114 |
torch2PIL = transforms.ToPILImage()
|
| 115 |
output = torch2PIL(output)
|
| 116 |
return output
|