Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ _loaded_models = {}
|
|
| 6 |
tab_state = "single"
|
| 7 |
def load_model(model_name):
|
| 8 |
if model_name not in _loaded_models:
|
| 9 |
-
_loaded_models[model_name] = SentenceTransformer(model_name)
|
| 10 |
return _loaded_models[model_name]
|
| 11 |
|
| 12 |
# ----------------------------- Core Functions -----------------------------
|
|
@@ -61,6 +61,7 @@ with gr.Blocks(
|
|
| 61 |
label="Select Model",
|
| 62 |
choices=[
|
| 63 |
"Tarka-AIR/Tarka-Embedding-150M-V1",
|
|
|
|
| 64 |
"sentence-transformers/all-MiniLM-L6-v2",
|
| 65 |
"intfloat/e5-base-v2",
|
| 66 |
"BAAI/bge-small-en-v1.5"
|
|
@@ -106,9 +107,10 @@ with gr.Blocks(
|
|
| 106 |
tarka_selector = gr.Dropdown(
|
| 107 |
label="Tarka Model",
|
| 108 |
choices=[
|
|
|
|
| 109 |
"Tarka-AIR/Tarka-Embedding-150M-V1",
|
| 110 |
],
|
| 111 |
-
value="Tarka-AIR/Tarka-Embedding-
|
| 112 |
)
|
| 113 |
|
| 114 |
open_selector = gr.Dropdown(
|
|
|
|
| 6 |
tab_state = "single"
|
| 7 |
def load_model(model_name):
|
| 8 |
if model_name not in _loaded_models:
|
| 9 |
+
_loaded_models[model_name] = SentenceTransformer(model_name,trust_remote_code=True)
|
| 10 |
return _loaded_models[model_name]
|
| 11 |
|
| 12 |
# ----------------------------- Core Functions -----------------------------
|
|
|
|
| 61 |
label="Select Model",
|
| 62 |
choices=[
|
| 63 |
"Tarka-AIR/Tarka-Embedding-150M-V1",
|
| 64 |
+
"Tarka-AIR/Tarka-Embedding-350M-V1",
|
| 65 |
"sentence-transformers/all-MiniLM-L6-v2",
|
| 66 |
"intfloat/e5-base-v2",
|
| 67 |
"BAAI/bge-small-en-v1.5"
|
|
|
|
| 107 |
tarka_selector = gr.Dropdown(
|
| 108 |
label="Tarka Model",
|
| 109 |
choices=[
|
| 110 |
+
"Tarka-AIR/Tarka-Embedding-350M-V1",
|
| 111 |
"Tarka-AIR/Tarka-Embedding-150M-V1",
|
| 112 |
],
|
| 113 |
+
value="Tarka-AIR/Tarka-Embedding-350M-V1"
|
| 114 |
)
|
| 115 |
|
| 116 |
open_selector = gr.Dropdown(
|