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 |
|