Spaces:
Build error
Build error
File size: 614 Bytes
48f1ac0 420a54e 48f1ac0 420a54e 9e59d28 48f1ac0 420a54e 48f1ac0 420a54e 48f1ac0 420a54e 48f1ac0 420a54e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
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)
|