Update app.py

app.py

@@ -28,12 +28,66 @@ models = [
 
 
 ####  REM-BG
+
+model_choices = keys(remove_bg_models)
+
 remove_bg_models = {
     "TracerUniversalB7": "TracerUniversalB7",
     "U2NET": "u2net",
     "U2NET Human Seg": "u2net_human_seg",
     "U2NET Cloth Seg": "u2net_cloth_seg"
+
+def predict(image, session, smoot, matting, bg_color):
+
+    session = new_session(remove_bg_models[session])
+
+    try:
+        return remove(session, image, smoot, matting, bg_color)
+    except ValueError as err:
+        logging.error(err)
+        return make_label(str(err)), None
+
+
+def change_show_mask(chk_state):
+    return gr.Image.update(visible=chk_state)
+
+
+def change_include_matting(chk_state):
+    return gr.Box.update(visible=chk_state), (0, 0, 0), 0, 0, 0
+
+
+def change_foreground_threshold(fg_value, value):
+    fg, bg, erode = value
+    return fg_value, bg, erode
+
+
+def change_background_threshold(bg_value, value):
+    fg, bg, erode = value
+    return fg, bg_value, erode
+
+
+def change_erode_size(erode_value, value):
+    fg, bg, erode = value
+    return fg, bg, erode_value
+
+
+def set_dominant_color(chk_state):
+    return chk_state, gr.ColorPicker.update(value=False, visible=not chk_state)
+
+
+def change_picker_color(picker, dominant):
+    if not dominant:
+        return picker
+    return dominant
+
+
+def change_background_mode(chk_state):
+    return gr.ColorPicker.update(value=False, visible=chk_state), \
+        gr.Checkbox.update(value=False, visible=chk_state)
+
+    
 }
+
 ###########
 
 text_gen = gr.Interface.load("spaces/daspartho/prompt-extend")