nickmuchi commited on
Commit
9855b70
·
1 Parent(s): ef03b1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -30,11 +30,13 @@ def load_vectorstore(model):
30
  embed_instruction='Represent the Financial paragraph for retrieval: ')
31
  return FAISS.load_local('vanguard-embeddings-inst', emb)
32
 
 
33
  vectorstore = load_vectorstore(model_options['all-mpnet-base-v2'])
34
 
35
  def on_value_change(change):
 
36
  global vectorstore
37
- vectorstore = load_vectostore(model_options[change])
38
 
39
  # vectorstore = load_vectorstore('vanguard-embeddings',sbert_emb)
40
 
@@ -127,7 +129,7 @@ with block:
127
  with gr.Row():
128
  embeddings = gr.Radio(choices=model_options_list,value=model_options_list[0], label='Choose your Embedding Model',
129
  interactive=True)
130
- embeddings.observe(on_value_change, names=model_options_list[0])
131
 
132
  print(embeddings.value)
133
 
 
30
  embed_instruction='Represent the Financial paragraph for retrieval: ')
31
  return FAISS.load_local('vanguard-embeddings-inst', emb)
32
 
33
+ #default embeddings
34
  vectorstore = load_vectorstore(model_options['all-mpnet-base-v2'])
35
 
36
  def on_value_change(change):
37
+ '''When radio changes, change the embeddings'''
38
  global vectorstore
39
+ vectorstore = load_vectorstore(model_options[change])
40
 
41
  # vectorstore = load_vectorstore('vanguard-embeddings',sbert_emb)
42
 
 
129
  with gr.Row():
130
  embeddings = gr.Radio(choices=model_options_list,value=model_options_list[0], label='Choose your Embedding Model',
131
  interactive=True)
132
+ embeddings.observe(on_value_change, names='value')
133
 
134
  print(embeddings.value)
135