rahulnamdev commited on
Commit
d5ff2ff
·
verified ·
1 Parent(s): 89cf407

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def predict_price_tool(arg1:str)-> float: #it's import to specify the return typ
14
  Args:
15
  arg1: the category of product
16
  """
17
- filter_dataset = dataset['train']['Category'].unique()
18
  #filter_dataset_min = filter_dataset['Selling Price'].min()
19
  #filter_dataset_max = filter_dataset['Selling Price'].min()
20
  return 1
 
14
  Args:
15
  arg1: the category of product
16
  """
17
+ filter_dataset = set(dataset['train']['Category'])
18
  #filter_dataset_min = filter_dataset['Selling Price'].min()
19
  #filter_dataset_max = filter_dataset['Selling Price'].min()
20
  return 1