Reeve commited on
Commit
74aec6d
1 Parent(s): 0ecdffa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ pretransform = transforms.Compose([
16
  transforms.ToTensor(),
17
  transforms.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])])
18
 
19
- def pipeline(img, v):
20
  img = model_build.img_preprocess(img, pretransform)
21
  with torch.no_grad():
22
  latent_space = psp(img.float(), randomize_noise=True, resize=False, return_latents=True)
 
16
  transforms.ToTensor(),
17
  transforms.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])])
18
 
19
+ def pipeline(img):
20
  img = model_build.img_preprocess(img, pretransform)
21
  with torch.no_grad():
22
  latent_space = psp(img.float(), randomize_noise=True, resize=False, return_latents=True)