Update app.py

app.py

@@ -4,7 +4,7 @@ from image_wgan import ImageWgan
 import os
 from os.path import exists
 def init():
-    generated_samples_folder = "results64"
+    generated_samples_folder = "/"
     discriminator_saved_model = "discriminator64.model"
     generator_saved_model = "generator64.model"
     latent_space = 100
@@ -24,7 +24,7 @@ def crop():
     import generator
     res = 64
     if res != 0:
-        results = "results64/generated.png"
+        results = "generated.png"
         img = Image.open(results)
 
         width,height = img.size
@@ -38,7 +38,8 @@ def crop():
             imgcrop = img.crop((left,top,left+res,res+2))
 
 
-            imgcrop.save("results64/"+str(i)+".png")        
+            imgcrop.save(str(i)+".png")     
+           
 init()
 
 
@@ -49,8 +50,8 @@ init()
 def gen():
     init()
     crop()
-    img = Image.open("results64/0.png")
-    print(img)
+    img = Image.open("0.png")
+
     return "hi"
 
 iface = gr.Interface(fn=gen, inputs=None, outputs="text")