nickmuchi commited on
Commit
d19a03c
1 Parent(s): 8425393

Update pages/2_Twitter_GPT_Search.py

Browse files
Files changed (1) hide show
  1. pages/2_Twitter_GPT_Search.py +4 -1
pages/2_Twitter_GPT_Search.py CHANGED
@@ -63,6 +63,9 @@ search_input = st.text_input(
63
  sbert_model_name = st.sidebar.selectbox("Embedding Model", options=list(bi_enc_dict.keys()), key='sbox')
64
 
65
  tweets = st.session_state['tlist']
 
 
 
66
 
67
  try:
68
 
@@ -74,7 +77,7 @@ try:
74
  text=f"Loading {model} embedding model and Generating Response..."
75
  ):
76
 
77
- vectorstore = create_vectorstore(tweets,model)
78
 
79
  tweets = embed_tweets(search_input,prompt,vectorstore)
80
 
 
63
  sbert_model_name = st.sidebar.selectbox("Embedding Model", options=list(bi_enc_dict.keys()), key='sbox')
64
 
65
  tweets = st.session_state['tlist']
66
+ topic = st.session_state['topic']
67
+ user = st.session_state['username']
68
+ cr_time = st.session_state['creation_time']
69
 
70
  try:
71
 
 
77
  text=f"Loading {model} embedding model and Generating Response..."
78
  ):
79
 
80
+ vectorstore = create_vectorstore(tweets,model,user,topic,cr_time)
81
 
82
  tweets = embed_tweets(search_input,prompt,vectorstore)
83