viettqse150044 commited on
Commit
f2c0e4b
·
1 Parent(s): 6aee03b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,7 +13,7 @@ def load_image(image_file):
13
  def streamlit_app():
14
  detection_model_path = "weight_files/clothes_detection_model.pt"
15
  background_model_path = "weight_files/model.h5"
16
- # save_path = ""
17
  image_file = None
18
  st.title("""WELCOME TO MY APP""")
19
  st.subheader("""FOR BACKGROUND REMOVAL AND CHANGE!""")
@@ -64,7 +64,7 @@ def streamlit_app():
64
  st.write('\n')
65
  st.write('\n')
66
 
67
- file_name = final_img_path.split("\\")[-1].split(".")[-2] + ".png"
68
  if st.download_button(
69
  label="Download postprocessing image",
70
  data=file,
 
13
  def streamlit_app():
14
  detection_model_path = "weight_files/clothes_detection_model.pt"
15
  background_model_path = "weight_files/model.h5"
16
+ save_path = ""
17
  image_file = None
18
  st.title("""WELCOME TO MY APP""")
19
  st.subheader("""FOR BACKGROUND REMOVAL AND CHANGE!""")
 
64
  st.write('\n')
65
  st.write('\n')
66
 
67
+ file_name = save_path.split("\\")[-1].split(".")[-2] +"from_abc" + ".png"
68
  if st.download_button(
69
  label="Download postprocessing image",
70
  data=file,