import streamlit as st
from transformers import pipeline
import pandas as pd

tqa = pipeline(task="table-question-answering", 
                    model="google/tapas-base-finetuned-wtq")

st.title("Table Question Answering")

file_name = st.file_uploader("Upload dataset",type=['csv','xlsx'])

if file_name is not None:
  try:
      df=pd.read_csv(file_name)
  except:
      df = pd.read_excel(file_name)
  df = df.astype(str)

  question = st.text_input('Type your question')
  with st.spinner():
    if(st.button('Answer')):
      answer = tqa(table=df, query=question,truncation=True)
      st.success(answer)