Update app.py
Browse files
app.py
CHANGED
@@ -29,13 +29,15 @@ models = [
|
|
29 |
|
30 |
#### REM-BG
|
31 |
|
32 |
-
model_choices = keys(remove_bg_models)
|
33 |
-
|
34 |
remove_bg_models = {
|
35 |
"TracerUniversalB7": "TracerUniversalB7",
|
36 |
"U2NET": "u2net",
|
37 |
"U2NET Human Seg": "u2net_human_seg",
|
38 |
"U2NET Cloth Seg": "u2net_cloth_seg"
|
|
|
|
|
|
|
|
|
39 |
|
40 |
def predict(image, session, smoot, matting, bg_color):
|
41 |
|
@@ -85,8 +87,7 @@ def change_background_mode(chk_state):
|
|
85 |
return gr.ColorPicker.update(value=False, visible=chk_state), \
|
86 |
gr.Checkbox.update(value=False, visible=chk_state)
|
87 |
|
88 |
-
|
89 |
-
}
|
90 |
|
91 |
###########
|
92 |
|
|
|
29 |
|
30 |
#### REM-BG
|
31 |
|
|
|
|
|
32 |
remove_bg_models = {
|
33 |
"TracerUniversalB7": "TracerUniversalB7",
|
34 |
"U2NET": "u2net",
|
35 |
"U2NET Human Seg": "u2net_human_seg",
|
36 |
"U2NET Cloth Seg": "u2net_cloth_seg"
|
37 |
+
}
|
38 |
+
|
39 |
+
model_choices = keys(remove_bg_models)
|
40 |
+
|
41 |
|
42 |
def predict(image, session, smoot, matting, bg_color):
|
43 |
|
|
|
87 |
return gr.ColorPicker.update(value=False, visible=chk_state), \
|
88 |
gr.Checkbox.update(value=False, visible=chk_state)
|
89 |
|
90 |
+
|
|
|
91 |
|
92 |
###########
|
93 |
|