ZennyKenny commited on
Commit
349af26
Β·
verified Β·
1 Parent(s): 63eaef6

fix update_columns

Browse files
Files changed (1) hide show
  1. app.py +10 -6
app.py CHANGED
@@ -122,22 +122,26 @@ def train_model(dataset_id, custom_dataset_id, label_column, feature_columns,
122
 
123
  return output_text, fig, columns_available
124
 
125
- def update_columns(dataset_id, custom_dataset_id):
126
  """
127
- Callback to dynamically fetch the columns from the dataset
128
- so the user can pick which columns to use as features/labels.
129
  """
130
  if dataset_id != "SKIP/ENTER_CUSTOM":
131
  final_id = dataset_id
 
132
  else:
 
133
  final_id = custom_dataset_id.strip()
 
134
 
135
- # Try to load the dataset and return columns
136
  try:
137
- ds = load_dataset(final_id, split="train")
 
 
 
138
  df = pd.DataFrame(ds)
139
  cols = df.columns.tolist()
140
- # Return as list of selectable options
141
  return gr.update(choices=cols), gr.update(choices=cols), f"Columns found: {cols}"
142
  except Exception as e:
143
  return gr.update(choices=[]), gr.update(choices=[]), f"Error loading {final_id}: {e}"
 
122
 
123
  return output_text, fig, columns_available
124
 
125
+ def update_columns(dataset_id, dataset_config, custom_dataset_id):
126
  """
127
+ Load the dataset from HF hub, using either the suggested one or the custom user-specified,
128
+ plus an optional config.
129
  """
130
  if dataset_id != "SKIP/ENTER_CUSTOM":
131
  final_id = dataset_id
132
+ final_config = dataset_config.strip() if dataset_config else None
133
  else:
134
+ # Use the user-supplied text
135
  final_id = custom_dataset_id.strip()
136
+ final_config = None # or parse from text if you like
137
 
 
138
  try:
139
+ if final_config:
140
+ ds = load_dataset(final_id, final_config, split="train")
141
+ else:
142
+ ds = load_dataset(final_id, split="train")
143
  df = pd.DataFrame(ds)
144
  cols = df.columns.tolist()
 
145
  return gr.update(choices=cols), gr.update(choices=cols), f"Columns found: {cols}"
146
  except Exception as e:
147
  return gr.update(choices=[]), gr.update(choices=[]), f"Error loading {final_id}: {e}"