Update app.py

app.py

@@ -23,7 +23,6 @@ create = st.button("Create The Model")
 
 if create:
     st.session_state.t2m_mod = create_model(loc=the_type)
-    st.session_state.generator = torch.Generator("cpu").manual_seed(int(bu_1))
 
 prom = st.text_input("# Prompt",'')
 
@@ -41,6 +40,8 @@ with c4:
 with c5:
   sl_2 = st.slider("hight",128,1024,512,8)
 
+st.session_state.generator = torch.Generator("cpu").manual_seed(int(bu_1))
+
 create = st.button("Imagine")
 
 if create: