Update app.py

app.py

@@ -16,13 +16,15 @@ def create_model(loc = "stabilityai/stable-diffusion-2-1-base", mch = 'cpu'):
 # - `FIRST RUN OF THE CODE`
 # - `CHANGING MODEL`""")
 
-the_type = st.selectbox("Model",("stabilityai/stable-diffusion-2-1-base",
-                                      "CompVis/stable-diffusion-v1-4"))
+# the_type = st.selectbox("Model",("stabilityai/stable-diffusion-2-1-base",
+#                                       "CompVis/stable-diffusion-v1-4"))
 
-create = st.button("Create The Model")
+# create = st.button("Create The Model")
 
-if create:
-    st.session_state.t2m_mod = create_model(loc=the_type)
+# if create:
+#     st.session_state.t2m_mod = create_model(loc=the_type)
+
+st.session_state.t2m_mod = create_model(loc=the_type)
 
 
 prom = st.text_input("Prompt",'')