Spaces:
Runtime error
Runtime error
root
commited on
Commit
·
58bc3a9
1
Parent(s):
450482a
add img choice
Browse files- ui_functions.py +1 -0
ui_functions.py
CHANGED
@@ -95,6 +95,7 @@ def copy_img_to_input(img, idx):
|
|
95 |
"图片3(img4)":2,
|
96 |
"图片4(img5)":3,
|
97 |
}
|
|
|
98 |
image_data = re.sub('^data:image/.+;base64,', '', img[idx])
|
99 |
processed_image = Image.open(BytesIO(base64.b64decode(image_data)))
|
100 |
tab_update = gr.update(selected='img2img_tab')
|
|
|
95 |
"图片3(img4)":2,
|
96 |
"图片4(img5)":3,
|
97 |
}
|
98 |
+
idx = idx_map[idx]
|
99 |
image_data = re.sub('^data:image/.+;base64,', '', img[idx])
|
100 |
processed_image = Image.open(BytesIO(base64.b64decode(image_data)))
|
101 |
tab_update = gr.update(selected='img2img_tab')
|