feat(web): support uploading file in model info

This commit is contained in:
源文雨 2024-06-04 21:29:48 +09:00
parent ee72eca6a5
commit 8ed0dec181
2 changed files with 3 additions and 2 deletions

View File

@ -1536,8 +1536,8 @@ with gr.Blocks(title="RVC WebUI") as app:
)
with gr.Row():
with gr.Column():
ckpt_path1 = gr.Textbox(
label=i18n("模型路径"), value="", interactive=True
ckpt_path1 = gr.File(
label=i18n("模型路径")
)
but8 = gr.Button(i18n("查看"), variant="primary")
with gr.Column():

View File

@ -63,6 +63,7 @@ def show_model_info(cpt, show_long_id=False):
def show_info(path):
try:
if hasattr(path, "name"): path = path.name
a = torch.load(path, map_location="cpu")
txt = show_model_info(a, show_long_id=True)
del a