simone-papicchio franceth commited on
Commit
45d1f9d
·
verified ·
1 Parent(s): ca89877

Fix models selection on p-np metrics (#18)

Browse files

- Fix models selection on p-np metrics (55c2a96d993529afddb1253d3ace37436c7bb86e)


Co-authored-by: Francesco Giannuzzo <[email protected]>

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -937,6 +937,7 @@ with gr.Blocks(theme='shivi/calm_seafoam', css_paths='style.css', js=js_func) as
937
 
938
  if input_data["input_method"]=="default":
939
  df = pd.read_csv(pnp_path)
 
940
  df['model'] = df['model'].replace('DeepSeek-R1-Distill-Llama-70B', 'DS-Llama3 70B')
941
  df['model'] = df['model'].replace('gpt-3.5', 'GPT-3.5')
942
  df['model'] = df['model'].replace('gpt-4o-mini', 'GPT-4o-mini')
 
937
 
938
  if input_data["input_method"]=="default":
939
  df = pd.read_csv(pnp_path)
940
+ df = df[df['model'].isin(input_data["models"])]
941
  df['model'] = df['model'].replace('DeepSeek-R1-Distill-Llama-70B', 'DS-Llama3 70B')
942
  df['model'] = df['model'].replace('gpt-3.5', 'GPT-3.5')
943
  df['model'] = df['model'].replace('gpt-4o-mini', 'GPT-4o-mini')