MINGYISU commited on
Commit
badf9ba
·
1 Parent(s): 1fe53bf

fixed bugs

Browse files
Files changed (1) hide show
  1. utils.py +5 -7
utils.py CHANGED
@@ -110,7 +110,7 @@ def create_hyperlinked_names(df):
110
  return row
111
 
112
  df = df.copy()
113
- df = df.apply(add_link_to_model_name)
114
  return df
115
 
116
  def fetch_data(file: str) -> pd.DataFrame:
@@ -122,15 +122,13 @@ def fetch_data(file: str) -> pd.DataFrame:
122
  response = requests.get(url)
123
  if response.status_code != 200:
124
  raise requests.HTTPError(f"Failed to fetch data: HTTP status code {response.status_code}")
125
- return pd.read_csv(io.StringIO(response.text))
126
-
127
- def get_urls(file: str='urls.csv') -> dict:
128
- urls = fetch_data(file)
129
- return dict(zip(urls['Models'], urls['URL']))
130
 
131
  def get_df(file="results.jsonl"):
132
  df = fetch_data(file)
133
- df.to_json(file, orient='records', lines=True)
 
 
134
  df['Model Size(B)'] = df['Model Size(B)'].apply(process_model_size)
135
  df = df.sort_values(by=['Overall'], ascending=False)
136
  df = create_hyperlinked_names(df)
 
110
  return row
111
 
112
  df = df.copy()
113
+ df = df.apply(add_link_to_model_name, axis=1)
114
  return df
115
 
116
  def fetch_data(file: str) -> pd.DataFrame:
 
122
  response = requests.get(url)
123
  if response.status_code != 200:
124
  raise requests.HTTPError(f"Failed to fetch data: HTTP status code {response.status_code}")
125
+ return pd.read_json(io.StringIO(response.text), orient='records', lines=True)
 
 
 
 
126
 
127
  def get_df(file="results.jsonl"):
128
  df = fetch_data(file)
129
+ print(df.columns)
130
+ print('URL' in df.columns)
131
+ print(df)
132
  df['Model Size(B)'] = df['Model Size(B)'].apply(process_model_size)
133
  df = df.sort_values(by=['Overall'], ascending=False)
134
  df = create_hyperlinked_names(df)