File size: 516 Bytes
a5b2f6f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from transformers import pipeline
import pandas as pd

# prepare table + question
data = {"Neighborhood": ["Upper East Side", "Soho", "Upper West Side"], "Number of Apartments": ["87", "53", "69"]}
table = pd.DataFrame.from_dict(data)
question = "how many apartments does Upper East Side have?"

# pipeline model
# Note: you must to install torch-scatter first.
tqa = pipeline(task="table-question-answering", model="google/tapas-large-finetuned-wtq")

# result

print(tqa(table=table, query=query)['cells'][0])
#53