Update app.py
Browse files
app.py
CHANGED
@@ -12,6 +12,8 @@ from threading import Lock
|
|
12 |
model_options = {'all-mpnet-base-v2': "sentence-transformers/all-mpnet-base-v2",
|
13 |
'instructor-base': "hkunlp/instructor-base"}
|
14 |
|
|
|
|
|
15 |
def load_vectorstore(model):
|
16 |
'''load embeddings and vectorstore'''
|
17 |
|
@@ -114,7 +116,8 @@ with block:
|
|
114 |
)
|
115 |
|
116 |
with gr.Row():
|
117 |
-
embeddings = gr.Radio(choices=
|
|
|
118 |
print(embeddings)
|
119 |
|
120 |
vectorstore = load_vectorstore(model_options[embeddings])
|
|
|
12 |
model_options = {'all-mpnet-base-v2': "sentence-transformers/all-mpnet-base-v2",
|
13 |
'instructor-base': "hkunlp/instructor-base"}
|
14 |
|
15 |
+
model_options_list = list(model_options.keys())
|
16 |
+
|
17 |
def load_vectorstore(model):
|
18 |
'''load embeddings and vectorstore'''
|
19 |
|
|
|
116 |
)
|
117 |
|
118 |
with gr.Row():
|
119 |
+
embeddings = gr.Radio(choices=model_options_list,value=model_options_list[0], label='Choose your Embedding Model',
|
120 |
+
interactive=True)
|
121 |
print(embeddings)
|
122 |
|
123 |
vectorstore = load_vectorstore(model_options[embeddings])
|