atrytone commited on
Commit
4bc6ef5
1 Parent(s): bd2c51e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -96,7 +96,7 @@ def inference(query):
96
  return [a_output, j_output, n_output]
97
 
98
 
99
- model_name = "biodatlab/MIReAD-Neuro-Contrastive"
100
  model_kwargs = {'device': 'cpu'}
101
  encode_kwargs = {'normalize_embeddings': False}
102
  faiss_embedder = HuggingFaceEmbeddings(
@@ -105,7 +105,7 @@ faiss_embedder = HuggingFaceEmbeddings(
105
  encode_kwargs=encode_kwargs
106
  )
107
 
108
- vecdb = FAISS.load_local("medline_contr", faiss_embedder)
109
 
110
 
111
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
96
  return [a_output, j_output, n_output]
97
 
98
 
99
+ model_name = "biodatlab/MIReAD-Neuro-Large"
100
  model_kwargs = {'device': 'cpu'}
101
  encode_kwargs = {'normalize_embeddings': False}
102
  faiss_embedder = HuggingFaceEmbeddings(
 
105
  encode_kwargs=encode_kwargs
106
  )
107
 
108
+ vecdb = FAISS.load_local("medline_index", faiss_embedder)
109
 
110
 
111
  with gr.Blocks(theme=gr.themes.Soft()) as demo: