fix: add binarize_model

app.py

@@ -18,7 +18,8 @@ example_images = [
 tasks = {
     "unwrap": ["UVDoc", None],
     "unshadow": ["GCDnet",  None],
-    "unblur": ["OpenCvBilateral", "NAFDPM", None]
+    "unblur": ["OpenCvBilateral", "NAFDPM", None],
+    "binarize": ["UnetCnn", None]
 }
 
 def process_image(img_path, unwrap_model, unshadow_model, binarize_model, unblur_model):
@@ -80,7 +81,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)
+                binarize_dropdown = gr.Dropdown(label="Select Binarize Model", choices=tasks["binarize"], value=None)
                 unblur_dropdown = gr.Dropdown(label="Select Unblur Model", choices=tasks["unblur"], value="OpenCvBilateral")
 
                 run_button = gr.Button("summit")