[email protected] commited on
Commit
b005996
·
1 Parent(s): 94bb3a0

link for proprietary models

Browse files
Files changed (1) hide show
  1. src/display/formatting.py +7 -0
src/display/formatting.py CHANGED
@@ -3,6 +3,13 @@ def model_hyperlink(link, model_name):
3
 
4
 
5
  def make_clickable_model(model_name):
 
 
 
 
 
 
 
6
  link = f"https://huggingface.co/{model_name}"
7
  return model_hyperlink(link, model_name)
8
 
 
3
 
4
 
5
  def make_clickable_model(model_name):
6
+ proprietary_models = {
7
+ 'o3-mini': 'https://openai.com/index/openai-o3-mini/',
8
+ 'o1': 'https://openai.com/o1/',
9
+ 'o1-preview': 'https://openai.com/index/introducing-openai-o1-preview/'
10
+ }
11
+ if model_name in proprietary_models:
12
+ return proprietary_models[model_name]
13
  link = f"https://huggingface.co/{model_name}"
14
  return model_hyperlink(link, model_name)
15