Update app.py

app.py

@@ -14,11 +14,11 @@ def predict_price_tool(arg1:str)-> float: #it's import to specify the return typ
     Args:
         arg1: the category of product
     """
-    filter_dataset = dataset['train']['Category' == arg1]
-    filter_dataset_min = filter_dataset['Selling Price'].min()
-    filter_dataset_max = filter_dataset['Selling Price'].min()
-
-    return (filter_dataset_min + filter_dataset_max )/2
+    filter_dataset = dataset['train']['Category'].unique()
+    #filter_dataset_min = filter_dataset['Selling Price'].min()
+    #filter_dataset_max = filter_dataset['Selling Price'].min()
+    return 1
+    #return (filter_dataset_min + filter_dataset_max )/2
 
 
 #Agent Example