Ankush05 commited on
Commit
cd43b43
1 Parent(s): 4025e7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(message, candidate_labels=["Reminder", "General Conversation"])
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"):