Spaces:
Configuration error
Configuration error
Muhammad Rama Nurimani
commited on
Commit
·
cfbe563
1
Parent(s):
1515d09
test deploy, bismillah
Browse files
app.py
CHANGED
@@ -21,7 +21,7 @@ def style_transfer(input_image):
|
|
21 |
transforms.Grayscale(num_output_channels=1),
|
22 |
transforms.Resize((256, 256)),
|
23 |
transforms.ToTensor(),
|
24 |
-
transforms.Normalize(mean=[0.5],
|
25 |
])
|
26 |
input_tensor = transform(input_image).unsqueeze(0)
|
27 |
with torch.no_grad():
|
|
|
21 |
transforms.Grayscale(num_output_channels=1),
|
22 |
transforms.Resize((256, 256)),
|
23 |
transforms.ToTensor(),
|
24 |
+
transforms.Normalize(mean=[0.5], std=[0.5])
|
25 |
])
|
26 |
input_tensor = transform(input_image).unsqueeze(0)
|
27 |
with torch.no_grad():
|