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