taskswithcode
commited on
Commit
•
2f0a627
1
Parent(s):
67cf80d
Fixes
Browse files
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 = {
|
|
|
|
|
|
|
|
|
|
|
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"] = "
|
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()
|