Spaces:
Runtime error
Runtime error
Мясников Филипп Сергеевич
commited on
Commit
•
e3b2423
1
Parent(s):
ceabeea
Fix
Browse files
app.py
CHANGED
@@ -132,10 +132,10 @@ def inference(img, model):
|
|
132 |
ffhq_codes = ffhq_codes + ffhq_latent_avg.repeat(ffhq_codes.shape[0], 1, 1)
|
133 |
|
134 |
cat_codes = cat_encoder(transformed_image.unsqueeze(0).to(device).float())
|
135 |
-
cat_codes = cat_codes +
|
136 |
|
137 |
dog_codes = dog_encoder(transformed_image.unsqueeze(0).to(device).float())
|
138 |
-
dog_codes = dog_codes +
|
139 |
|
140 |
npimage = gen_im(ffhq_codes, dog_codes, cat_codes, model)
|
141 |
|
|
|
132 |
ffhq_codes = ffhq_codes + ffhq_latent_avg.repeat(ffhq_codes.shape[0], 1, 1)
|
133 |
|
134 |
cat_codes = cat_encoder(transformed_image.unsqueeze(0).to(device).float())
|
135 |
+
cat_codes = cat_codes + cat_latent_avg.repeat(cat_codes.shape[0], 1, 1)
|
136 |
|
137 |
dog_codes = dog_encoder(transformed_image.unsqueeze(0).to(device).float())
|
138 |
+
dog_codes = dog_codes + dog_latent_avg.repeat(dog_codes.shape[0], 1, 1)
|
139 |
|
140 |
npimage = gen_im(ffhq_codes, dog_codes, cat_codes, model)
|
141 |
|