File size: 1,007 Bytes
fc824ef 9c34a20 fc824ef d9e9300 fc824ef 69aae83 9c34a20 ffa852c fc824ef 9379371 fc824ef 9379371 fc824ef 9379371 fc824ef da5f5c8 10d5462 9c34a20 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
# AnjiBot 🤖
### Computer Science Course Representative (Chatbot)
from transformers import pipeline, TFTapasForQuestionAnswering, TapasTokenizer
import pandas as pd
import gradio as gr
import tensorflow_probability
# Load TAPAS model and table
model = TFTapasForQuestionAnswering.from_pretrained("google/tapas-base-finetuned-sqa")
tokenizer = TapasTokenizer.from_pretrained("google/tapas-base-finetuned-sqa")
tqa = pipeline(task="table-question-answering", model=model, tokenizer=tokenizer)
table = pd.read_csv('CSLECTURERS.csv')
table = table.astype('str')
messages = []
responses = []
def anjibot(message, history):
messages.append(message)
conversation = {"text": message, "past_user_input": messages, "generated_responses": responses}
answer = tqa(table=table, query=message)["answer"]
responses.append(answer)
return "AnjiBot: " + answer
chatbot = gr.ChatInterface(anjibot, title='AnjiBot', description="Anji is unavailable? That girl! Ask me, I may know!")
chatbot.launch() |