Update app.py
Browse files
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()}
|