File size: 664 Bytes
74c96e6
143fe54
 
74c96e6
143fe54
74c96e6
143fe54
 
 
 
 
 
 
74c96e6
143fe54
 
 
 
 
74c96e6
 
143fe54
 
74c96e6
 
 
 
 
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
29
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()