Update app.py

app.py

@@ -133,7 +133,12 @@ def generate(
             fn_index=3    
         )
         result = job.result()
-        list_dir(result)
+
+        for root, dirs, files in os.walk(result):
+            for file in files:
+                if file.endswith('.jpg'):
+                    images.append(os.path.json(root,file))
+                    
     return images, seed
 
 examples = [