DINGOLANI commited on
Commit
3f671f4
·
verified ·
1 Parent(s): 9373e7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -18,12 +18,12 @@ def get_column_name(possible_names, df):
18
  for name in possible_names:
19
  if name in df.columns:
20
  return name
21
- raise KeyError(f"None of the expected column names {possible_names} found in dataset")
22
 
23
  # Map column names dynamically
24
- designer_column = get_column_name(["Designer", "Brand"], df)
25
- category_column = get_column_name(["Category"], df)
26
- product_column = get_column_name(["Product", "Product Name"], df)
27
 
28
  # Load full dataset
29
  df = pd.read_csv(csv_file, nrows=10000)
 
18
  for name in possible_names:
19
  if name in df.columns:
20
  return name
21
+ raise KeyError(f"None of the expected column names {possible_names} found in dataset. Available columns: {df.columns}")
22
 
23
  # Map column names dynamically
24
+ designer_column = get_column_name(["brand_name"], df)
25
+ category_column = get_column_name(["product_category"], df)
26
+ product_column = get_column_name(["product_name"], df)
27
 
28
  # Load full dataset
29
  df = pd.read_csv(csv_file, nrows=10000)