3laa2 commited on
Commit
b16a490
·
1 Parent(s): c3b349b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,11 +11,11 @@ def create_model(loc = "stabilityai/stable-diffusion-2-1-base", mch = 'cpu'):
11
  pipe = pipe.to(mch)
12
  return pipe
13
 
14
- t2i = st.checkbox("Text2Image")
15
 
16
  the_type = st.selectbox("Model Name",("stabilityai/stable-diffusion-2-1-base",
17
  "CompVis/stable-diffusion-v1-4"))
18
- create = st.checkbox("Create The Model")
19
 
20
  if create:
21
  st.title("Text2Image")
 
11
  pipe = pipe.to(mch)
12
  return pipe
13
 
14
+ t2i = st.title("#Text2Image")
15
 
16
  the_type = st.selectbox("Model Name",("stabilityai/stable-diffusion-2-1-base",
17
  "CompVis/stable-diffusion-v1-4"))
18
+ create = st.button("Create The Model")
19
 
20
  if create:
21
  st.title("Text2Image")