Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
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)
|