Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,8 @@ def create_model(loc = "stabilityai/stable-diffusion-2-1-base", mch = 'cpu'):
|
|
13 |
|
14 |
t2i = st.checkbox("Text2Image")
|
15 |
|
16 |
-
|
|
|
17 |
create = st.checkbox("Create The Model")
|
18 |
|
19 |
if create:
|
|
|
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:
|