Fix

app.py

@@ -132,10 +132,10 @@ def inference(img, model):
     ffhq_codes = ffhq_codes + ffhq_latent_avg.repeat(ffhq_codes.shape[0], 1, 1)
 
     cat_codes = cat_encoder(transformed_image.unsqueeze(0).to(device).float())
-    cat_codes = cat_codes + ffhq_latent_avg.repeat(cat_codes.shape[0], 1, 1)
+    cat_codes = cat_codes + cat_latent_avg.repeat(cat_codes.shape[0], 1, 1)
 
     dog_codes = dog_encoder(transformed_image.unsqueeze(0).to(device).float())
-    dog_codes = dog_codes + ffhq_latent_avg.repeat(dog_codes.shape[0], 1, 1)
+    dog_codes = dog_codes + dog_latent_avg.repeat(dog_codes.shape[0], 1, 1)
         
     npimage = gen_im(ffhq_codes, dog_codes, cat_codes, model)