Spaces:
Paused
Paused
Wenye He
commited on
Update app.py
Browse files
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"
|
| 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 |
-
|
| 119 |
-
["
|
| 120 |
-
value="
|
| 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 |
|