from transformers import pipeline import pandas as pd # prepare table + question data = {"Actors": ["Brad Pitt", "Leonardo Di Caprio", "George Clooney"], "Number of movies": ["87", "53", "69"]} table = pd.DataFrame.from_dict(data) question = "how many movies does Leonardo Di Caprio 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