HugSib commited on
Commit
5df3253
·
verified ·
1 Parent(s): 3c64e23

Delete utils.py

Browse files
Files changed (1) hide show
  1. utils.py +0 -37
utils.py DELETED
@@ -1,37 +0,0 @@
1
- from data.model_handler import ModelHandler
2
-
3
- def make_clickable_model(model_name, link=None):
4
- if link is None:
5
- desanitized_model_name = model_name.replace("_", "/")
6
-
7
- if '/captioning' in desanitized_model_name:
8
- desanitized_model_name = desanitized_model_name.replace('/captioning', '')
9
- if '/ocr' in desanitized_model_name:
10
- desanitized_model_name = desanitized_model_name.replace('/ocr', '')
11
-
12
- link = "https://huggingface.co/" + desanitized_model_name
13
-
14
- return f'<a target="_blank" style="text-decoration: underline" href="{link}">{model_name}</a>'
15
-
16
-
17
- def add_rank_and_format(df):
18
- df = df.reset_index()
19
- df = df.rename(columns={"index": "Model"})
20
- df = ModelHandler.add_rank(df)
21
- df["Model"] = df["Model"].apply(make_clickable_model)
22
- return df
23
-
24
- def get_refresh_function():
25
- def _refresh(metric):
26
- model_handler = ModelHandler()
27
- data_task_category = model_handler.get_vidore_data(metric)
28
- df = add_rank_and_format(data_task_category)
29
- return df
30
-
31
- return _refresh
32
-
33
-
34
- def filter_models(data, search_term):
35
- if search_term:
36
- data = data[data['Model'].str.contains(search_term, case=False, na=False)]
37
- return data