wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-14 23:42:35 +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): def run_prompt(job):
data = {'images':[],'rainbow':"None"} data = {'images':[],'rainbow':"None",'lama':"None"}
#Inferring from the Rainbow Script #Inferring from the Rainbow Script
image_string = job['input']['image_string_lama'] image_string = job['input']['image_string_lama']
@ -98,6 +98,16 @@ def run_prompt(job):
new_working_directory = "../" new_working_directory = "../"
os.chdir(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 #NOTE This is for the Rainbow Script Will be updating it later
# if image_string != 'None': # if image_string != 'None':