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()
|