updating app

app.py

@@ -80,7 +80,6 @@ def main():
 
         else:
             st.session_state['text'] = txt
-            st.session_state['graph'] = None
 
             # preprocess
             sentence = ie.preprocess(txt)
@@ -107,6 +106,7 @@ def main():
 
             st.session_state['graph'] = graph
 
+    if st.session_state['graph'] is not None:
         st.divider()
         st.write("Does this prediction look correct?")
         collector.st_feedback(
@@ -121,6 +121,9 @@ def main():
             single_submit=False
         )
 
+        st.session_state['text'] = ''
+        st.session_state['graph'] = None
+
 
 if __name__ == '__main__':
     main()