Spaces:
Build error
Build error
Upload 2 files
Browse files- app.py +15 -12
- requirements.txt +3 -1
app.py
CHANGED
@@ -1,20 +1,23 @@
|
|
1 |
import streamlit as st
|
2 |
from transformers import pipeline
|
3 |
-
|
4 |
|
5 |
-
|
|
|
6 |
|
7 |
-
st.title("
|
8 |
|
9 |
-
file_name = st.file_uploader("Upload
|
10 |
|
11 |
if file_name is not None:
|
12 |
-
|
|
|
|
|
|
|
|
|
13 |
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
for p in predictions:
|
20 |
-
col2.subheader(f"{ p['label'] }: { round(p['score'] * 100, 1)}%")
|
|
|
1 |
import streamlit as st
|
2 |
from transformers import pipeline
|
3 |
+
import pandas as pd
|
4 |
|
5 |
+
tqa = pipeline(task="table-question-answering",
|
6 |
+
model="google/tapas-base-finetuned-wikisql-supervised")
|
7 |
|
8 |
+
st.title("Table Question Answering")
|
9 |
|
10 |
+
file_name = st.file_uploader("Upload dataset",type=['csv','xlsx'])
|
11 |
|
12 |
if file_name is not None:
|
13 |
+
try:
|
14 |
+
df=pd.read_csv(file_name)
|
15 |
+
except:
|
16 |
+
df = pd.read_excel(file_name)
|
17 |
+
df = df.astype(str)
|
18 |
|
19 |
+
question = st.text_input('Type your question')
|
20 |
+
with st.spinner():
|
21 |
+
if(st.button('Answer')):
|
22 |
+
answer = tqa(table=df, query=question,truncation=True)
|
23 |
+
st.success(answer)
|
|
|
|
requirements.txt
CHANGED
@@ -1,2 +1,4 @@
|
|
1 |
transformers
|
2 |
-
torch
|
|
|
|
|
|
1 |
transformers
|
2 |
+
torch
|
3 |
+
pandas
|
4 |
+
torch-scatter
|