Create models/onnx.py
Browse files- app/tabs/models/onnx.py +21 -0
app/tabs/models/onnx.py
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import gradio as gr
|
| 2 |
+
from original impoet *
|
| 3 |
+
|
| 4 |
+
|
| 5 |
+
def onnx_conv():
|
| 6 |
+
with gr.TabItem(i18n("Onnx导出")):
|
| 7 |
+
with gr.Row():
|
| 8 |
+
ckpt_dir = gr.Textbox(
|
| 9 |
+
label=i18n("RVC模型路径"), value="", interactive=True
|
| 10 |
+
)
|
| 11 |
+
with gr.Row():
|
| 12 |
+
onnx_dir = gr.Textbox(
|
| 13 |
+
label=i18n("Onnx输出路径"), value="", interactive=True
|
| 14 |
+
)
|
| 15 |
+
with gr.Row():
|
| 16 |
+
infoOnnx = gr.Label(label="info")
|
| 17 |
+
with gr.Row():
|
| 18 |
+
butOnnx = gr.Button(i18n("导出Onnx模型"), variant="primary")
|
| 19 |
+
butOnnx.click(
|
| 20 |
+
export_onnx, [ckpt_dir, onnx_dir], infoOnnx, api_name="export_onnx"
|
| 21 |
+
)
|