ovshake commited on
Commit
1646c1f
1 Parent(s): 1b1a677

fix gpu error

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -129,7 +129,7 @@ body_part = st.radio("Would you like to try clothes on your upper body (such as
129
  stable_fashion_args.part = body_part
130
  resolution = st.radio("Which resolution would you like to get the resulting picture in? (Keep in mind, higher the resolution, higher the queue times)", (128, 256, 512), index=2)
131
  stable_fashion_args.resolution = resolution
132
- rembg_status = st.radio("Would you like to remove background in your image before putting new clothes on you? (Sometimes it results in better images)", ("Yes", "No"), index=1)
133
  stable_fashion_args.rembg = (rembg_status == "Yes")
134
  guidance_scale = st.slider("Select a guidance scale. 7.5 gives the best results.", 1.0, 15.0, value=7.5)
135
  stable_fashion_args.guidance_scale = guidance_scale
 
129
  stable_fashion_args.part = body_part
130
  resolution = st.radio("Which resolution would you like to get the resulting picture in? (Keep in mind, higher the resolution, higher the queue times)", (128, 256, 512), index=2)
131
  stable_fashion_args.resolution = resolution
132
+ rembg_status = st.radio("Would you like to remove background in your image before putting new clothes on you? (Sometimes it results in better images)", ("Yes", "No"), index=0)
133
  stable_fashion_args.rembg = (rembg_status == "Yes")
134
  guidance_scale = st.slider("Select a guidance scale. 7.5 gives the best results.", 1.0, 15.0, value=7.5)
135
  stable_fashion_args.guidance_scale = guidance_scale