wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-10 05:22:34 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Updated test_input json

Browse Source
This commit is contained in:
jarvis2324 2023-08-09 19:15:24 +05:30
parent 819933e149
commit a325c92217
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
comfy_runpod.py
View File

@ -66,7 +66,7 @@ def get_images(ws, prompt):
def run_prompt(job):
data = {'images':[],'rainbow':"None"}
data = {'images':[],'rainbow':"None",'lama':"None"}
#Inferring from the Rainbow Script
image_string = job['input']['image_string_lama']
@ -97,6 +97,16 @@ def run_prompt(job):
new_working_directory = "../"
os.chdir(new_working_directory)
image = Image.open('./lama-with-refiner/output/')
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)
#NOTE This is for the Rainbow Script Will be updating it later