pikto commited on
Commit
b978214
·
1 Parent(s): 3bffb64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
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