3laa2 commited on
Commit
34bbbfe
·
1 Parent(s): 89dd0cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -11,14 +11,13 @@ def create_model(loc = "stabilityai/stable-diffusion-2-1-base", mch = 'cpu'):
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")
22
  t2m_mod = create_model(loc=the_type)
23
 
24
 
 
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
  t2m_mod = create_model(loc=the_type)
22
 
23