Update app.py

app.py

@@ -139,6 +139,7 @@ if uploaded_file is not None:
     st.markdown('</div>', unsafe_allow_html=True)
 
     button_clicked = st.button("상단의 두 사진을 딥페이크 모델에 학습시키기")
+    st.markdown('</div>', unsafe_allow_html=True)
 
     if button_clicked:
         st.markdown('<div class="container">', unsafe_allow_html=True)
@@ -166,3 +167,4 @@ if uploaded_file is not None:
 
 
         st.markdown('<div class="bottom-space"></div>', unsafe_allow_html=True)
+        st.markdown('</div>', unsafe_allow_html=True)