taskswithcode commited on
Commit
2f0a627
1 Parent(s): 67cf80d
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -15,8 +15,13 @@ from streamlit_image_select import image_select
15
 
16
 
17
 
18
- use_case = {"1":"Image background removal","2":"Masking foreground for downstream inpainting task"}
19
- mask_types = {"blur - blurs background":"blur","map - makes the foreground white and rest black ":"map","rgba - makes background white":"rgba","green - makes the background green":"green"}
 
 
 
 
 
20
 
21
 
22
 
@@ -134,7 +139,7 @@ def init_session():
134
  st.session_state["download_ready"] = None
135
  st.session_state["model_name"] = "ss_test"
136
  st.session_state["file_name"] = "default"
137
- st.session_state["mask_type"] = "blur"
138
 
139
  def app_main(app_mode,example_files,model_name_files,config_file):
140
  init_session()
 
15
 
16
 
17
 
18
+ use_case = {"1":"Image background removal - (upload any picture and remove background)","2":"Masking foreground for downstream inpainting task"}
19
+ mask_types = {
20
+ "rgba - makes background white":"rgba",
21
+ "green - makes the background green":"green",
22
+ "blur - blurs background":"blur",
23
+ "map - makes the foreground white and rest black ":"map"
24
+ }
25
 
26
 
27
 
 
139
  st.session_state["download_ready"] = None
140
  st.session_state["model_name"] = "ss_test"
141
  st.session_state["file_name"] = "default"
142
+ st.session_state["mask_type"] = "rgba"
143
 
144
  def app_main(app_mode,example_files,model_name_files,config_file):
145
  init_session()