Spaces:
Running
Running
| # This is based on https://huggingface.co/spaces/Abbasid/TableQA | |
| from transformers import pipeline | |
| import pandas as pd | |
| import gradio as gr | |
| tqa = pipeline(task="table-question-answering", model="google/tapas-large-finetuned-wtq") | |
| query = "How many points did Jaylen Brown score?" | |
| def main(filepath, query): | |
| tableau = pd.read_csv(filepath).astype(str) | |
| result = tqa(table=tableau, query=query)["answer"] | |
| return result | |
| iface = gr.Interface( | |
| fn=main, | |
| inputs=[ | |
| gr.File(type="filepath", label="Upload csv"), | |
| gr.Textbox(type="text", label="What do you want to know?"), | |
| ], | |
| outputs=[gr.Textbox(type="text", label="The output will be here" , lines=3)] | |
| ) | |
| # Launch the Gradio interface | |
| iface.launch() |