Upload folder using huggingface_hub

app.py

@@ -86,7 +86,7 @@ def predict_store_total_sales_batch():
     predicted_store_total_sales = [round(float(np.exp(log_total_sales)), 2) for log_total_sales in predicted_log_total_sales]
 
     # Create a dictionary of predictions with Product Id as Unique keys for each record
-    product_ids = input_data['id'].tolist()  # Assuming 'id' is the Product ID column
+    product_ids = input_data['Product_Id'].tolist()  # Assuming 'id' is the Product ID column
     output_dict = dict(zip(product_ids, predicted_store_total_sales))  # Use actual prices
 
     # Return the predictions dictionary as a JSON response