From 30ff122f84e2dc4d81c15c590415601c09051859 Mon Sep 17 00:00:00 2001 From: Hendrik Langer Date: Mon, 3 Jul 2023 22:52:49 +0200 Subject: [PATCH] kandinsky2 endpoint --- matrix_pygmalion_bot/bot/wrappers/runpod.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/matrix_pygmalion_bot/bot/wrappers/runpod.py b/matrix_pygmalion_bot/bot/wrappers/runpod.py index 65305b3..132edaf 100644 --- a/matrix_pygmalion_bot/bot/wrappers/runpod.py +++ b/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