Muhammad Rama Nurimani commited on
Commit
cfbe563
·
1 Parent(s): 1515d09

test deploy, bismillah

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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], 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():