Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -391,7 +391,13 @@ with gr.Blocks() as demo:
|
|
391 |
|
392 |
with gr.Row():
|
393 |
input2 = gr.Textbox(label="自定义文件名")
|
394 |
-
output_type = gr.Dropdown(
|
|
|
|
|
|
|
|
|
|
|
|
|
395 |
),
|
396 |
shape0_str = gr.Textbox(label="shape0 (逗号分隔的整数)", value="1,3,128,128")
|
397 |
shape1_str = gr.Textbox(label="shape1 (逗号分隔的整数)", value="0,0,0,0")
|
|
|
391 |
|
392 |
with gr.Row():
|
393 |
input2 = gr.Textbox(label="自定义文件名")
|
394 |
+
output_type = gr.Dropdown(
|
395 |
+
choices=["TorchScript", "ONNX", "Fixed", "MNN", "NCNN"],
|
396 |
+
value=["TorchScript", "ONNX", "MNN", "ONNX"],
|
397 |
+
multiselect=True,
|
398 |
+
label="模型类型",
|
399 |
+
info="1. 生成mnn和ncnn模型必须先生成onnx模型;2.如果选项中包含了Fixed,那么输出的onnx和mnn模型都使用固定shape的input。"
|
400 |
+
)
|
401 |
),
|
402 |
shape0_str = gr.Textbox(label="shape0 (逗号分隔的整数)", value="1,3,128,128")
|
403 |
shape1_str = gr.Textbox(label="shape1 (逗号分隔的整数)", value="0,0,0,0")
|