Jaswanth-0821 commited on
Commit
a24182d
·
verified ·
1 Parent(s): 02f59f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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-150M-V1"
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(