From f91733b911eb5e319d2e3a4c0bf28cfe00363f03 Mon Sep 17 00:00:00 2001 From: jarvis2324 Date: Wed, 9 Aug 2023 19:38:01 +0530 Subject: [PATCH] Updated Comfy --- comfy_runpod.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy_runpod.py b/comfy_runpod.py index b6117cf3c..a2a1eb90f 100644 --- a/comfy_runpod.py +++ b/comfy_runpod.py @@ -101,13 +101,13 @@ def run_prompt(job): new_working_directory = "../" os.chdir(new_working_directory) - image = Image.open('./lama-with-refiner/output/') + image = Image.open('./lama-with-refiner/output/image_mask.png') im_file = io.BytesIO() image.save(im_file, format="JPEG") im_bytes = im_file.getvalue() # im_bytes: image in binary format. im_b64 = base64.b64encode(im_bytes) im_b64 = str(im_b64) - data['images'].append(im_b64) + data['lama'].append(im_b64)