Spaces:

cthleen
/

batik-test2

Sleeping

cthleen commited on Apr 7, 2025

Commit

8f4be28

1 Parent(s): 07be559

fix error

Files changed (1) hide show

stylegan_generator.py CHANGED Viewed

@@ -28,6 +28,6 @@ def generate_stylegan():
     image = (image * 0.5 + 0.5) * 255
     image = image.astype(np.uint8)
     image = np.transpose(image, (1, 2, 0))
-    pil_image = Image.fromarray(image, 'RGB')
     return pil_img.resize((512, 512), Image.LANCZOS)

     image = (image * 0.5 + 0.5) * 255
     image = image.astype(np.uint8)
     image = np.transpose(image, (1, 2, 0))
+    pil_img = Image.fromarray(image, 'RGB')
     return pil_img.resize((512, 512), Image.LANCZOS)