import gradio as gr
import torch
from transformers import pipeline
import pandas as pd

tqa = pipeline(task = 'table-question-answering',model='google/tapas-base-finetuned-wtq')
table = pd.read_csv("queries.csv")
table = table.astype(str)
def input_query(Type_your_query_here):
  return (tqa(table=table,query=Type_your_query_here)['answer'])

iface = gr.Interface(fn=input_query,inputs="text",outputs="text", title = "Siva's Assistant",description="My assistant will answer the queries related to our section's Faculty Name,Faculty Mobile Number,Faculity Cabin Number,Classroom Number,Timetable of the class for each subject")  

iface.launch()