Update app.py

app.py

@@ -3,7 +3,7 @@ from image_dataset import ImageDataset
 from image_wgan import ImageWgan
 import os
 from os.path import exists
-from PIL import Image
+from PIL import Image, ImageEnhance
 import numpy
 def init():
     generated_samples_folder = "."
@@ -58,8 +58,12 @@ def gen(seed):
     img1 = Image.open("1.png")
     img2 = Image.open("2.png")
     img3 = Image.open("3.png")
-    
-    return img0, img1, img2, img3
+    img = []
+    for i in range(4):
+        img.append(Image.open(i+".png")
+        filter = ImageEnhance.Color(img[i])
+        img[i] = img[i].filter(3)
+    return img
 
 iface = gr.Interface(
     fn=gen,