Update app.py

app.py

@@ -54,15 +54,9 @@ def gen(seed):
     numpy.random.seed(int(seed))
     init()
     crop()
-    img0 = Image.open("0.png")
-    img1 = Image.open("1.png")
-    img2 = Image.open("2.png")
-    img3 = Image.open("3.png")
     imgArr = []
-    for i in range(3):
+    for i in range(4):
         imgArr.append(Image.open(str(i)+".png"))
-        
-    #return img0,img1,img2,img3
     return imgArr
 
 iface = gr.Interface(