Update app.py

app.py

@@ -144,15 +144,15 @@ if upload_file is not None:
     file_extension = upload_file.name.split('.')[-1].lower()
     if file_extension == 'csv':
         try:
-            df1 = pd.read_csv(upload_file, na_filter=False)
-            if df1.isnull().values.any():
+            df = pd.read_csv(upload_file, na_filter=False)
+            if df.isnull().values.any():
                 st.error("Error: The CSV file contains missing values.")
                 st.stop()
             else:
-                st.dataframe(df1, key="csv_dataframe")
-                st.write("_number of rows_", df1.shape[0])
-                st.write("_number of columns_", df1.shape[1])
-                st.session_state.df1 = df1  
+                st.dataframe(df, key="csv_dataframe")
+                st.write("_number of rows_", df.shape[0])
+                st.write("_number of columns_", df.shape[1])
+                st.session_state.df = df  
         except pd.errors.ParserError:
             st.error("Error: The CSV file is not readable or is incorrectly formatted.")
             st.stop()
@@ -201,7 +201,7 @@ if st.button("Retrieve your answer"):
         time.sleep(5)          
         if df1 is not None:
             tqa = pipeline(task="table-question-answering", model="microsoft/tapex-large-finetuned-wtq")
-            st.write(tqa(table=df1, query=question)['answer'])
+            st.write(tqa(table=df, query=question)['answer'])
 
 st.divider()
 st.write(f"Number of questions asked: {st.session_state['question_attempts']}/{max_attempts}")