Spaces:

bankholdup
/

stylegan_petbreeder

Runtime error

Мясников Филипп Сергеевич commited on May 18, 2022

Commit

d36b237

1 Parent(s): b3de1a8

fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -126,7 +126,7 @@ def gen_im(model_type='ffhq'):
 def inference(img):
     img.save('out.jpg')
     #aligned_face = align_face('out.jpg')
-    input_image = run_alignment(image_path)
     transformed_image = transform(input_image)
     ffhq_codes = ffhq_encoder(transformed_image.unsqueeze(0).to(device).float())

 def inference(img):
     img.save('out.jpg')
     #aligned_face = align_face('out.jpg')
+    input_image = run_alignment('out.jpg')
     transformed_image = transform(input_image)
     ffhq_codes = ffhq_encoder(transformed_image.unsqueeze(0).to(device).float())