awacke1 commited on
Commit
63ad468
1 Parent(s): 248dfb1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -167,7 +167,7 @@ with gr.Blocks(theme = gr.themes.Soft()) as demo:
167
  return gr.Textbox.update(visible=False)
168
 
169
  past_searches.change(update_past_search_content, past_searches, past_search_content)
170
-
171
  def update_with_rag_md(message, llm_results_use = 5, database_choice = index_info, llm_model_picked = 'mistralai/Mistral-7B-Instruct-v0.2'):
172
  prompt_text_from_data = ""
173
  database_to_use = database_choice
@@ -181,12 +181,12 @@ with gr.Blocks(theme = gr.themes.Soft()) as demo:
181
  arxiv_search_success = False
182
  except:
183
  arxiv_search_success = False
184
-
185
  if not arxiv_search_success:
186
  gr.Warning("Arxiv Search not working, switching to semantic search ...")
187
  rag_out = get_rag(message)
188
  database_to_use = index_info
189
-
190
  md_text_updated = mark_text
191
  for i in range(retrieve_results):
192
  rag_answer = rag_out[i]
@@ -198,8 +198,8 @@ with gr.Blocks(theme = gr.themes.Soft()) as demo:
198
  md_text_updated += md_text_paper
199
  prompt = get_prompt_text(message, prompt_text_from_data, llm_model_picked = llm_model_picked)
200
  save_search_results(message, md_text_updated)
201
- return md_text_updated, prompt, gr.Dropdown.update(choices=get_past_searches())
202
-
203
  def ask_llm(prompt, llm_model_picked = 'mistralai/Mistral-7B-Instruct-v0.2', stream_outputs = False):
204
  model_disabled_text = "LLM Model is disabled"
205
  output = ""
 
167
  return gr.Textbox.update(visible=False)
168
 
169
  past_searches.change(update_past_search_content, past_searches, past_search_content)
170
+
171
  def update_with_rag_md(message, llm_results_use = 5, database_choice = index_info, llm_model_picked = 'mistralai/Mistral-7B-Instruct-v0.2'):
172
  prompt_text_from_data = ""
173
  database_to_use = database_choice
 
181
  arxiv_search_success = False
182
  except:
183
  arxiv_search_success = False
184
+
185
  if not arxiv_search_success:
186
  gr.Warning("Arxiv Search not working, switching to semantic search ...")
187
  rag_out = get_rag(message)
188
  database_to_use = index_info
189
+
190
  md_text_updated = mark_text
191
  for i in range(retrieve_results):
192
  rag_answer = rag_out[i]
 
198
  md_text_updated += md_text_paper
199
  prompt = get_prompt_text(message, prompt_text_from_data, llm_model_picked = llm_model_picked)
200
  save_search_results(message, md_text_updated)
201
+ return md_text_updated, prompt, get_past_searches()
202
+
203
  def ask_llm(prompt, llm_model_picked = 'mistralai/Mistral-7B-Instruct-v0.2', stream_outputs = False):
204
  model_disabled_text = "LLM Model is disabled"
205
  output = ""