fix: add binarize_model

app.py

@@ -21,12 +21,14 @@ tasks = {
     "unblur": ["OpenCvBilateral", "NAFDPM", None]
 }
 
-def process_image(img_path, unwrap_model, unshadow_model, unblur_model):
+def process_image(img_path, unwrap_model, unshadow_model, binarize_model, unblur_model):
     task_list = []
     if unwrap_model:
         task_list.append(("unwrap", unwrap_model))
     if unshadow_model:
         task_list.append(("unshadow", unshadow_model))
+    if binarize_model:
+        task_list.append(("binarize", binarize_model))
     if unblur_model:
         task_list.append(("unblur", unblur_model))
 
@@ -78,6 +80,7 @@ def main():
 
                 unwrap_dropdown = gr.Dropdown(label="Select Unwrap Model", choices=tasks["unwrap"], value="UVDoc")
                 unshadow_dropdown = gr.Dropdown(label="Select Unshadow Model", choices=tasks["unshadow"], value="GCDnet")
+                binarize_dropdown = gr.Dropdown(label="Select Unshadow Model", choices=tasks["binarize"], value=None)
                 unblur_dropdown = gr.Dropdown(label="Select Unblur Model", choices=tasks["unblur"], value="OpenCvBilateral")
 
                 run_button = gr.Button("summit")
@@ -86,7 +89,7 @@ def main():
                 output_image = gr.Image(label="output")
                 elapse_textbox = gr.Textbox(label="Elapsed Time", interactive=False)
         # 绑定按钮点击事件
-        run_button.click(fn=process_image, inputs=[img_input, unwrap_dropdown, unshadow_dropdown, unblur_dropdown],outputs=[output_image, elapse_textbox])
+        run_button.click(fn=process_image, inputs=[img_input, unwrap_dropdown, unshadow_dropdown, binarize_dropdown, unblur_dropdown],outputs=[output_image, elapse_textbox])
 
     # 启动应用
     demo.launch()