File size: 829 Bytes
fc824ef d9e9300 fc824ef |
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 |
# AnjiBot 🤖
### Computer Science Course Representative (Chatbot)
from transformers import pipeline
import pandas as pd
import gradio as gr
import tensorflow_probability
# Load TAPAS model and table
tqa = pipeline(task="table-question-answering", model="google/tapas-base-finetuned-sqa")
table = pd.read_csv('CSLECTURERS.csv', delimiter="::")
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
demo_chatbot = gr.ChatInterface(anjibot, title='AnjiBot', description="Anji is unavailable? That girl! Ask me, I may know!")
demo_chatbot.launch() |