franceth commited on
Commit
55c2a96
·
verified ·
1 Parent(s): ca89877

Fix models selection on p-np metrics

Browse files
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')