EXR

app.py

@@ -35,7 +35,8 @@ def get_base64(bin_file):
 # set_background('ipad_bg.png')
 
 upload_img = (st.file_uploader("输入character sheet", "png", accept_multiple_files=True))
-upload_img2 = (st.file_uploader("pose images", "png", accept_multiple_files=True))
+pose_format = "exr"
+upload_img2 = (st.file_uploader("pose images", pose_format, accept_multiple_files=True))
 
 # os.system("sh download.sh")
 
@@ -46,7 +47,7 @@ if st.button("RUN!"):
     if upload_img is not None:
         os.makedirs("character_sheet", exist_ok=True)
         for i in range(len(upload_img)):
-            with open("character_sheet/{}.png".format(i), "wb") as f:
+            with open(f"character_sheet/{i}.{pose_format}", "wb") as f:
                 f.write(upload_img[i].read())
         os.makedirs("poses", exist_ok=True)
         for i, e in enumerate(sorted(upload_img2, key=lambda x: x.name)):