Update main.py

main.py

@@ -76,12 +76,12 @@ if "testing" not in st.session_state:
 
 with col_2:
     st.header("TEST THE PROMPT")
+    testing_prompt = st.text_area("Input your testing prompt")
     if st.session_state["testing"]:
         im = generate_image(pipe, testing_prompt)
         st.session_state["testing"] = False
     else:
         im = np.zeros([h,w,3])
-    testing_prompt = st.text_area("Input your testing prompt")
     st.session_state["testing"] = st.button("test the prompt")
     st.image(im)