Update app.py

app.py

@@ -88,8 +88,11 @@ def predict(choice, seed):
     else:
         z = torch.randn(64, 100, 1, 1)
         punks = model(z)
-        save_image(punks, "punks.png", normalize=True)
-        return 'punks.png'
+        save_image(punks, "image.png", normalize=True)
+        img = Image.open(f"image.png").convert('RGBA')
+        img.putalpha(255)
+        img.save("image.png")
+        return 'image.png'
 
 
 gr.Interface(