Upload 17 files

app.py

@@ -77,7 +77,7 @@ with st.sidebar:
     use_bm25 = st.checkbox("Use 2-Stage Retrieval (BM25)")
     num_candidates = int(
         st.number_input(
-            "Number of Candidates to Generate after first stage",
+            "Number of Candidates to Generate:",
             25,
             200,
             step=25,
@@ -171,10 +171,7 @@ if ticker_year_quarter_tuples_list != []:
     multi_doc_context = generate_multi_doc_context(context_group)
 
 else:
-    if use_bm25 == True:
-        indices = get_indices_bm25(data, None, None, None, num_candidates)
-    else:
-        indices = None
+    indices = None
 
     query_results = query_pinecone(
         dense_query_embedding,
@@ -205,11 +202,11 @@ if decoder_model == "GPT-3.5 Turbo":
                 value="",
                 type="password",
             )
-        gpt_submitted = st.form_submit_button("Submit")
-        if gpt_submitted:
-            api_key = save_key(openai_key)
-            openai.api_key = api_key
-            generated_text = gpt_turbo_model(edited_prompt)
+            gpt_submitted = st.form_submit_button("Submit")
+            if gpt_submitted:
+                api_key = save_key(openai_key)
+                openai.api_key = api_key
+                generated_text = gpt_turbo_model(edited_prompt)
 
 if decoder_model == "Vicuna-7B":
     with col2: