Files changed (1) hide show
  1. utils.py +10 -1
utils.py CHANGED
@@ -104,11 +104,20 @@ MODEL_URLS = {
104
  "CLIP-ViT-H-14-laion2B-s32B-b79K": "https://huggingface.co/laion/CLIP-ViT-H-14-laion2B-s32B-b79K",
105
  "e5-v": "https://huggingface.co/royokong/e5-v",
106
  "Magiclens": "https://github.com/google-deepmind/magiclens",
107
- "MMRet-large": "https://huggingface.co/JUNJIE99/MMRet-large"
 
 
 
108
  }
109
 
110
  def create_hyperlinked_names(df):
111
  def add_link_to_model_name(model_name):
 
 
 
 
 
 
112
  if model_name in MODEL_URLS:
113
  return f"<a href='{MODEL_URLS[model_name]}'>{model_name}</a>"
114
  else:
 
104
  "CLIP-ViT-H-14-laion2B-s32B-b79K": "https://huggingface.co/laion/CLIP-ViT-H-14-laion2B-s32B-b79K",
105
  "e5-v": "https://huggingface.co/royokong/e5-v",
106
  "Magiclens": "https://github.com/google-deepmind/magiclens",
107
+ "MMRet-large": "https://huggingface.co/JUNJIE99/MMRet-large",
108
+ "VLM2Vec-Phi-3.5-v": "https://huggingface.co/TIGER-Lab/VLM2Vec-Full",
109
+ "VLM2Vec": "https://github.com/TIGER-AI-Lab/VLM2Vec",
110
+ "UniIR": "https://huggingface.co/TIGER-Lab/UniIR"
111
  }
112
 
113
  def create_hyperlinked_names(df):
114
  def add_link_to_model_name(model_name):
115
+ if "VLM2Vec (Phi-3.5-V-" in model_name:
116
+ return f"<a href='{MODEL_URLS["VLM2Vec-Phi-3.5-v"]}'>{model_name}</a>"
117
+ if "VLM2Vec (LLaVA-1.6-LoRA-" in model_name:
118
+ return f"<a href='{MODEL_URLS["VLM2Vec"]}'>{model_name}</a>"
119
+ if "UniIR" in model_name:
120
+ return f"<a href='{MODEL_URLS["UniIR"]}'>{model_name}</a>"
121
  if model_name in MODEL_URLS:
122
  return f"<a href='{MODEL_URLS[model_name]}'>{model_name}</a>"
123
  else: