Fix

app.py

@@ -112,7 +112,7 @@ def run_alignment(image_path):
   return aligned_image
 
 
-def gen_im(model_type='ffhq'):
+def gen_im(ffhq_codes, dog_codes, cat_codes, model_type='ffhq'):
   if model_type=='ffhq':
     imgs, _ = ffhq_decoder([ffhq_codes], input_is_latent=True, randomize_noise=False, return_latents=True)
   elif model_type=='dog':
@@ -140,7 +140,7 @@ def inference(img):
     dog_codes = dog_codes + ffhq_latent_avg.repeat(dog_codes.shape[0], 1, 1)
         
     animal = "cat"
-    npimage = gen_im(animal)
+    npimage = gen_im(ffhq_codes, dog_codes, cat_codes, animal)
 	
     imageio.imwrite('filename.jpeg', npimage)
     return 'filename.jpeg'