InfoDeskBot / app.py
dr-wasim's picture
updated code
143fe54
raw
history blame contribute delete
664 Bytes
import gradio as gr
import csv
from transformers import pipeline
responses = {}
with open('Answers.csv', mode='r', encoding='utf-8-sig') as file:
reader = csv.DictReader(file)
for row in reader:
intent = row['Intent']
answer = row['Answer']
responses[intent] = answer
def respond(message, history):
pipe = pipeline("text-classification", model="sehrishsafdar/urdu-intent-model")
intent = pipe.predict(message)
print(intent)
return responses[intent[0]['label']]
demo = gr.ChatInterface(
respond,
title="Urdu University Information Desk Chatbot",
)
if __name__ == "__main__":
demo.launch()