Update app.py

app.py

@@ -123,9 +123,9 @@ if st.button("Retrieve your answer"):
     
     with st.spinner("Wait for it...", show_time=True):
         time.sleep(5)          
-        if df is not None:
+        if df_original is not None:
             tqa = pipeline(task="table-question-answering", model="microsoft/tapex-large-finetuned-wtq")
-            st.write(tqa(table=df, query=question)['answer'])
+            st.write(tqa(table=df_original, query=question)['answer'])
 
 st.divider()
 st.write(f"Number of questions asked: {st.session_state['question_attempts']}/{max_attempts}")