Nikhil0987 commited on
Commit
c226dff
1 Parent(s): 29616c7
Files changed (2) hide show
  1. chat.py +23 -23
  2. home.py +1 -1
chat.py CHANGED
@@ -1,26 +1,26 @@
1
- from transformers import pipeline, Conversation
2
- # import streamlit_option_menu
3
- import streamlit as st
4
-
5
- def Chat():
6
-
7
- query = st.chat_input("Enter your query")
8
- convo = pipeline("conversational")
9
- oracle = pipeline(task="zero-shot-classification", model="facebook/bart-large-mnli")
10
- usrinput = Conversation(query)
11
- chitchat = convo(usrinput)
12
- ans = oracle(
13
- query,
14
- candidate_labels=["logout"])
15
-
16
- if ans["scores"][0] > 0.85:
17
- st.session_state["user"] = "visitor"
18
- with st.chat_message("assistant"):
19
- "You are now living in dream"
20
- st.experimental_rerun()
21
- else:
22
- with st.chat_message("assistant"):
23
- chitchat
24
 
25
 
26
 
 
1
+ # from transformers import pipeline, Conversation
2
+ # # import streamlit_option_menu
3
+ # import streamlit as st
4
+
5
+ # def Chat():
6
+
7
+ # query = st.chat_input("Enter your query")
8
+ # convo = pipeline("conversational")
9
+ # oracle = pipeline(task="zero-shot-classification", model="facebook/bart-large-mnli")
10
+ # usrinput = Conversation(query)
11
+ # chitchat = convo(usrinput)
12
+ # ans = oracle(
13
+ # query,
14
+ # candidate_labels=["logout"])
15
+
16
+ # if ans["scores"][0] > 0.85:
17
+ # st.session_state["user"] = "visitor"
18
+ # with st.chat_message("assistant"):
19
+ # "You are now living in dream"
20
+ # st.experimental_rerun()
21
+ # else:
22
+ # with st.chat_message("assistant"):
23
+ # chitchat
24
 
25
 
26
 
home.py CHANGED
@@ -14,7 +14,7 @@ def dashboard():
14
  # st.header("Dashboard")
15
 
16
  with st.sidebar:
17
- selected = option_menu("Menu", ["Home", "Dashboard","Visual Q&A","Remainder","Conversation","Logout"], icons=['house', 'activity'], menu_icon="cast", default_index=0)
18
  if selected == "Home":
19
  homepage()
20
 
 
14
  # st.header("Dashboard")
15
 
16
  with st.sidebar:
17
+ selected = option_menu("Menu", ["Home", "Dashboar","Remainder","VISUAL Q&A","Conversation","Logout"], icons=['house', 'activity'], menu_icon="cast", default_index=0)
18
  if selected == "Home":
19
  homepage()
20