Мясников Филипп Сергеевич commited on
Commit
b3de1a8
1 Parent(s): 025efa1
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -126,8 +126,8 @@ def gen_im(model_type='ffhq'):
126
  def inference(img):
127
  img.save('out.jpg')
128
  #aligned_face = align_face('out.jpg')
129
- input_image = run_alignment(image_path)
130
- transformed_image = transform(input_image)
131
 
132
  ffhq_codes = ffhq_encoder(transformed_image.unsqueeze(0).to(device).float())
133
  ffhq_codes = ffhq_codes + ffhq_latent_avg.repeat(ffhq_codes.shape[0], 1, 1)
 
126
  def inference(img):
127
  img.save('out.jpg')
128
  #aligned_face = align_face('out.jpg')
129
+ input_image = run_alignment(image_path)
130
+ transformed_image = transform(input_image)
131
 
132
  ffhq_codes = ffhq_encoder(transformed_image.unsqueeze(0).to(device).float())
133
  ffhq_codes = ffhq_codes + ffhq_latent_avg.repeat(ffhq_codes.shape[0], 1, 1)