Update app.py

app.py

@@ -85,6 +85,11 @@ def apply_style(style_name: str, positive: str) -> Tuple[str]:
     p = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
     return p.replace("{prompt}", positive)
 
+def list_dir(path):
+    for root, dirs, files in os.walk(path):
+        for file in files:
+            print(os.path.join(root, file))
+
 def generate(
     prompt: str,
     seed: float = 0,
@@ -128,13 +133,7 @@ def generate(
             fn_index=3    
         )
         result = job.result()
-        for file in os.listdir(result):
-            print(file)
-            if file != 'captions.json':
-                img_path = os.path.join(result, file)
-                with open(img_path, 'r') as f:
-                    print(f)
-                    images.append((f))
+        list_dir(result)
     return images, seed
 
 examples = [