Update app.py

app.py

@@ -153,6 +153,7 @@ def main():
                 st.session_state.rewrite_a = ''
                 st.session_state.rewrite_b = ''
                 st.session_state.vote_submitted = False
+                st.session_state.rewrites_generated = False
 
                 with st.spinner("Генерация новых рерайтов..."):
 
@@ -168,7 +169,9 @@ def main():
                         st.session_state.model_a, st.session_state.original_review)
                     st.session_state.rewrite_b = generate_rewrite_for_model(
                         st.session_state.model_b, st.session_state.original_review)
-                st.experimental_rerun()
+                    st.session_state.rewrites_generated = True
+                    
+                st.rerun()
 
 if __name__ == '__main__':
     main()