nickmuchi commited on
Commit
7f4ceb4
·
1 Parent(s): d456856

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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=list(model_options.keys()),value=list(model_options.keys())[0], label='Choose your Embedding Model')
 
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])