Update app.py

app.py

@@ -130,12 +130,9 @@ def generate(
         result = job.result()
         for file in os.listdir(result):
             print(file)
-        json_file = os.path.join(result,"captions.json")
-        with open(json_file, 'r') as f:
-            print(f)
-            json_data = f
-        for img in json_data["data"]:
-            images.append(img)
+            if file != 'captions.json':
+                img_path = os.path.join(result, file)
+                images.append(img_path)
     return images, seed
 
 examples = [