thon | |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM | |
import pandas as pd | |
tokenizer = AutoTokenizer.from_pretrained("microsoft/tapex-large-finetuned-wtq") | |
model = AutoModelForSeq2SeqLM.from_pretrained("microsoft/tapex-large-finetuned-wtq") | |
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?" |