kandinsky2 endpoint

matrix_pygmalion_bot/bot/wrappers/runpod.py

@@ -147,10 +147,10 @@ class RunpodImageWrapper2(RunpodWrapper):
 
         os.makedirs("./.data/images", exist_ok=True)
         files = []
-        for image in output:
+        for image in output['images']:
             temp_name = next(tempfile._get_candidate_names())
             filename = "./.data/images/" + temp_name + ".jpg"
-            await self.download_image(image["image"], filename)
+            await self.download_image(image, filename)
             files.append(filename)
 
         return files