Wenye He commited on
Commit
c2d846d
·
verified ·
1 Parent(s): 7167bd9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -30,7 +30,7 @@ def load_vector_store(store_name):
30
  """Cache vector stores in memory"""
31
  if store_name not in vector_store_cache:
32
  vector_store_cache[store_name] = FAISS.load_local(
33
- f"vector_stores/{store_name}",
34
  embedder
35
  )
36
  return vector_store_cache[store_name]
@@ -115,9 +115,9 @@ with gr.Blocks() as demo:
115
  value="phi-3-mini",
116
  label="Select Model"
117
  )
118
- vector_store_dropdown = gr.Dropdown(
119
- ["legal_docs", "tech_docs"],
120
- value="tech_docs",
121
  label="Knowledge Base"
122
  )
123
 
 
30
  """Cache vector stores in memory"""
31
  if store_name not in vector_store_cache:
32
  vector_store_cache[store_name] = FAISS.load_local(
33
+ f"vector_store/{store_name}",
34
  embedder
35
  )
36
  return vector_store_cache[store_name]
 
115
  value="phi-3-mini",
116
  label="Select Model"
117
  )
118
+ vector_store_choice = gr.Dropdown(
119
+ ["llm", "scoliosis"],
120
+ value="scoliosis",
121
  label="Knowledge Base"
122
  )
123