maybe wild

app.py

@@ -1,3 +1,4 @@
+
 import streamlit as st
 import pandas as pd
 from transformers import pipeline
@@ -16,24 +17,26 @@ question = st.text_input("Enter your question:")
 
 # Process table and question
 if uploaded_file is not None and question:
-    # Read table from CSV
-    table = pd.read_csv(uploaded_file)
-    new_header = table.iloc[0] #grab the first row for the header
-    table = table[1:] #take the data less the header row
-    table.columns = new_header
+    try:
+        # Read table from CSV
+        table = pd.read_csv(uploaded_file)
+
+        # Display the table
+        st.write("Uploaded Table:")
+        st.dataframe(table)
 
-    # Display the table
-    st.write("Uploaded Table:")
-    st.write(table)
+        # Convert DataFrame to the format expected by TAPAS
+        table_data = table.to_dict(orient='records')
 
-    # Get answer
-    answer = tqa(table=table, query=question)['cells'][0]
+        # Get answer
+        answer = tqa(table=table_data, query=question)['cells'][0]
 
-    # Display the answer
-    st.write("Answer:", answer)
+        # Display the answer
+        st.write("Answer:", answer)
+    except Exception as e:
+        st.error(f"An error occurred: {e}")
 
 # Instructions
 st.markdown("""
-*First, upload a CSV file containing your table data. The CSV should have headers for each column.*  
-*Then, enter a question related to the table and press Enter to see the answer.*  
+*First, upload a CSV file.
 """)