Spaces:
Runtime error
Runtime error
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 | |