tumuyan2 commited on
Commit
6942279
·
1 Parent(s): 78ca519

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
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( ["TorchScript", "ONNX", "Fixed", "MNN", "NCNN"], value=["TorchScript", "ONNX", "MNN", "ONNX"], multiselect=True, label="模型类型", info="1. 生成mnn和ncnn模型必须先生成onnx模型;2.如果选项中包含了Fixed,那么输出的onnx和mnn模型都使用固定shape的input。"
 
 
 
 
 
 
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")