Update app.py
Browse files
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.
|
15 |
|
16 |
the_type = st.selectbox("Model Name",("stabilityai/stable-diffusion-2-1-base",
|
17 |
"CompVis/stable-diffusion-v1-4"))
|
18 |
-
create = st.
|
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")
|