Spaces:
Build error
Build error
import streamlit as st | |
from transformers import pipeline | |
import pandas as pd | |
tqa = pipeline(task="table-question-answering", | |
model="google/tapas-base-finetuned-wikisql-supervised") | |
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) | |