Upload folder using huggingface_hub

app.py

@@ -23,9 +23,11 @@ def predict_product_sales_price():
     sample = {
         'Product_Weight': product_data['Product_Weight'],
         'Product_Sugar_Content': product_data['Product_Sugar_Content'],
+        'Product_Allocated_Area': product_data['Product_Allocated_Area'],
         'Product_Type': product_data['Product_Type'],
         'Product_MRP': product_data['Product_MRP'],
         'Store_Id': product_data['Store_Id'],
+        'Store_Establishment_Year': product_data['Store_Establishment_Year'],
         'Store_Size': product_data['Store_Size'],
         'Store_Location_City_Type': product_data['Store_Location_City_Type'],
         'Store_Type': product_data['Store_Type']
@@ -43,12 +45,11 @@ def predict_product_sales_price():
 # Define an endpoint to predict product sales price for a batch of product
 @super_kart_api.post('/v1/productbatch')
 def predict_product_batch():
-   # Get the uploaded CSV file from the request
+    # Get the uploaded CSV file from the request
     file = request.files['file']
 
     # Read the file into a DataFrame
     input_data = pd.read_csv(file)
-    input_data = input_data.drop(['Product_Id','Store_Establishment_Year','Product_Allocated_Area'],axis=1)
 
     # Make predictions for the batch data
     predictions = model.predict(input_data).tolist()