Update app.py
Browse files
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("
|
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 |
|