Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ classifi = pipeline(model="facebook/bart-large-mnli")
|
|
17 |
def chatbot():
|
18 |
st.title("ChatBot")
|
19 |
if query := st.chat_input("Enter your message"):
|
20 |
-
ans = classifi(
|
21 |
if ans["labels"][0] == "Reminder":
|
22 |
values = getValues(query)
|
23 |
with st.chat_message("assistant"):
|
|
|
17 |
def chatbot():
|
18 |
st.title("ChatBot")
|
19 |
if query := st.chat_input("Enter your message"):
|
20 |
+
ans = classifi(query, candidate_labels=["Reminder", "General Conversation"])
|
21 |
if ans["labels"][0] == "Reminder":
|
22 |
values = getValues(query)
|
23 |
with st.chat_message("assistant"):
|