MINGYISU commited on
Commit
5897cc3
·
1 Parent(s): afc6d30

update urls

Browse files
Files changed (1) hide show
  1. utils.py +13 -4
utils.py CHANGED
@@ -111,11 +111,20 @@ MODEL_URLS = {
111
  }
112
 
113
  def create_hyperlinked_names(df):
 
 
 
114
  def add_link_to_model_name(model_name):
115
- if model_name in MODEL_URLS:
116
- return f"<a href='{MODEL_URLS[model_name]}'>{model_name}</a>"
117
- else:
118
- return model_name
 
 
 
 
 
 
119
 
120
  df = df.copy()
121
  df['Models'] = df['Models'].apply(add_link_to_model_name)
 
111
  }
112
 
113
  def create_hyperlinked_names(df):
114
+ def convert_url(url, model_name):
115
+ return f'<a href="{url}">{model_name}</a>'
116
+
117
  def add_link_to_model_name(model_name):
118
+ if "VLM2Vec (Phi-3.5-V-" in model_name:
119
+ url = MODEL_URLS["VLM2Vec-Phi-3.5-v"]
120
+ return convert_url(url, model_name)
121
+ if "VLM2Vec (LLaVA-1.6-LoRA-" in model_name:
122
+ url = MODEL_URLS["VLM2Vec"]
123
+ return convert_url(url, model_name)
124
+ if "UniIR" in model_name:
125
+ url = MODEL_URLS["UniIR"]
126
+ return convert_url(url, model_name)
127
+ return convert_url(MODEL_URLS[model_name], model_name) if model_name in MODEL_URLS else model_name
128
 
129
  df = df.copy()
130
  df['Models'] = df['Models'].apply(add_link_to_model_name)