change column names

app.py

@@ -52,6 +52,7 @@ example_df = pd.read_csv(file_name)
 
 def infer(num_sample):
     num_sample = int(num_sample)
+    col_dict = np.load("model/col_dict.npz", allow_pickle=True)['dict'].item()
     input = np.load("model/kde_white_female.npz", allow_pickle=True)['model'].item()
     kde, scaler, cols_names = input['model'], input['scaler'], input['columns']
     
@@ -60,8 +61,9 @@ def infer(num_sample):
     cov_list = np.array([[f'Synth_{i+1}', 'F', 'White'] for i in range(num_sample)])
     new_data = np.concatenate([cov_list, sample], axis=1)
     cols=['PTID','Sex','Race','Age']
-    cols.extend(fnmatch.filter(cols_names,'H_*'))
+    cols.extend([col_dict[i] for i in fnmatch.filter(cols_names,'H_*')])
     df_kde_synth = pd.DataFrame(new_data, columns=cols)
+    df_kde_synth['Age'] = round(df_kde_synth['Age'].astype('float'))
     df_kde_synth.to_csv(save_name, index=False)
     return gr.Dataframe(df_kde_synth.head(), label='Results (only showing the first few rows)', show_label=True), gr.Button("Download", link="/file="+save_name)