awinml commited on
Commit
0ba41da
1 Parent(s): a7b0635

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -171,7 +171,7 @@ elif encoder_model == "SGPT":
171
  retriever_model = get_sgpt_embedding_model()
172
 
173
 
174
- window = int(st.number_input("Sentence Window Size", 1, 3, value=1))
175
 
176
  threshold = float(
177
  st.number_input(
@@ -185,7 +185,7 @@ query_results = query_pinecone(
185
  query_text, num_results, retriever_model, pinecone_index, threshold
186
  )
187
 
188
- if threshold <= 0.65:
189
  context_list = sentence_id_combine(data, query_results, lag=window)
190
  else:
191
  context_list = format_query(query_results)
 
171
  retriever_model = get_sgpt_embedding_model()
172
 
173
 
174
+ window = int(st.number_input("Sentence Window Size", 0, 3, value=0))
175
 
176
  threshold = float(
177
  st.number_input(
 
185
  query_text, num_results, retriever_model, pinecone_index, threshold
186
  )
187
 
188
+ if threshold <= 0.60:
189
  context_list = sentence_id_combine(data, query_results, lag=window)
190
  else:
191
  context_list = format_query(query_results)