aidevhund commited on
Commit
e5a36e7
·
verified ·
1 Parent(s): f20c3fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -19,6 +19,12 @@ llm_display_names = {
19
  "meta-llama/Meta-Llama-3-8B-Instruct": "Meta-Llama-3",
20
  "mistralai/Mistral-7B-Instruct-v0.2": "Mistral-7B",
21
  }
 
 
 
 
 
 
22
 
23
  # Reverse mapping to retrieve original names
24
  llm_reverse_mapping = {v: k for k, v in llm_display_names.items()}
 
19
  "meta-llama/Meta-Llama-3-8B-Instruct": "Meta-Llama-3",
20
  "mistralai/Mistral-7B-Instruct-v0.2": "Mistral-7B",
21
  }
22
+ embed_models = [
23
+ "BAAI/bge-small-en-v1.5", # 33.4M
24
+ "NeuML/pubmedbert-base-embeddings",
25
+ "BAAI/llm-embedder", # 109M
26
+ "BAAI/bge-large-en" # 335M
27
+ ]
28
 
29
  # Reverse mapping to retrieve original names
30
  llm_reverse_mapping = {v: k for k, v in llm_display_names.items()}