Update app.py

app.py

@@ -197,7 +197,8 @@ def download_file2folder(url: str, folder: str, filesize_max: int, filesize_min:
             os.remove(save_path)
         return None
 
-async def _process_model(model_input: Union[str, gr.File], tilesize: int, output_dir: str,task_id:int,fp16:bool,onnxsim:bool,opset:int,dynamic_axes:bool):
+async def _process_model(model_input: Union[str, gr.File], tilesize: int, output_dir: str,task_id:int,fp16:bool,onnxsim:bool,opset:int,dynamic_axes:bool,fp16mnn:bool):
+
 
 
     log = ('初始化日志记录...\n')
@@ -244,7 +245,7 @@ async def _process_model(model_input: Union[str, gr.File], tilesize: int, output
     try:
         log += ( '正在将 ONNX 模型转换为 MNN 格式...\n')
         print_log(task_id, '正在将 ONNX 模型转换为 MNN 格式', '开始')
-        convertmnn(onnx_path, mnn_path,fp16)
+        convertmnn(onnx_path, mnn_path, fp16mnn)
         yield onnx_path,log
     except Exception as e:
         log += ( f'转换 MNN 模型时出错: {str(e)}\n')
@@ -280,9 +281,10 @@ with gr.Blocks() as demo:
             input_type.change(show_input, inputs=input_type, outputs=[url_input, file_input])
             tilesize = gr.Number(label="Dummy input width/height, default 64", value=64, precision=0)
             opset = gr.Number(label="ONNX export opset version, suggest 9/11/13/16/17/18", value=13, precision=0)
-            fp16 = gr.Checkbox(label="FP16", value=False)
+            fp16 = gr.Checkbox(label="Use fp16 in ONNX export", value=False)
             onnxsim = gr.Checkbox(label="ONNX export simplify model", value=False)
             dynamic_axes = gr.Checkbox(label="ONNX input apply dynamic axes", value=True)
+            fp16mnn = gr.Checkbox(label="Use fp16 in MNN convert, only reduce filesize", value=True)
             try_run = gr.Checkbox(label="MNNSR test", value=False)
             convert_btn = gr.Button("Run")
         with gr.Column():
@@ -298,9 +300,7 @@ with gr.Blocks() as demo:
                     return gr.update(visible=False)
             try_run.change(show_try_run, inputs=try_run, outputs=img_output)
 
-    async def process_model(input_type, url_input, file_input, tilesize, fp16, onnxsim, opset, dynamic_axes, try_run):
-
-
+    async def process_model(input_type, url_input, file_input, tilesize, fp16, onnxsim, opset, dynamic_axes, fp16mnn,try_run):
         global task_counter
         task_counter += 1
 
@@ -348,7 +348,8 @@ with gr.Blocks() as demo:
         onnx_path = None
         mnn_path = None
         # 调用重命名后的函数
-        async for result in _process_model(model_input, tilesize if tilesize>0 else 64, output_dir, task_counter, fp16, onnxsim, opset, dynamic_axes):
+        async for result in _process_model(model_input, tilesize if tilesize>0 else 64, output_dir, task_counter, fp16, onnxsim, opset, dynamic_axes, fp16mnn):
+
 
             if isinstance(result, tuple) and len(result) == 3:
                 onnx_path, mnn_path, process_log = result
@@ -372,17 +373,16 @@ with gr.Blocks() as demo:
 
     convert_btn.click(
         process_model,
-        inputs=[input_type, url_input, file_input, tilesize, fp16, onnxsim, opset, dynamic_axes, try_run],
+        inputs=[input_type, url_input, file_input, tilesize, fp16, onnxsim, opset, dynamic_axes, fp16mnn, try_run],
         outputs=[onnx_output, mnn_output, log_box, img_output],
         api_name="convert_nmm_model"
     )
 
-    # 将示例移至底部并包裹在列组件中
     examples_column = gr.Column(visible=True)
     with examples_column:
         examples = [
             [model_type_opt[0], "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-animevideov3.pth"],
-            [model_type_opt[0], "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth"],
+            [model_type_opt[0], "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.2.4/RealESRGAN_x4plus_anime_6B.pth"],
             [model_type_opt[0], "https://github.com/Phhofm/models/releases/download/4xNomos8kSC/4xNomos8kSC.pth"],
             [model_type_opt[0], "https://github.com/Phhofm/models/releases/download/1xDeJPG/1xDeJPG_SRFormer_light.pth"],
             [model_type_opt[0], "https://objectstorage.us-phoenix-1.oraclecloud.com/n/ax6ygfvpvzka/b/open-modeldb-files/o/4x-WTP-ColorDS.pth"],