Santhu976 commited on
Commit
ccb77ed
·
verified ·
1 Parent(s): 8f4cc7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -76,7 +76,7 @@ def predict_sales_total_batch():
76
  predicted_prices = model.predict(input_data).tolist()
77
 
78
  # Create a dictionary of predictions with property IDs as keys
79
- product_store_ids = input_data[['Product_Id', 'Store_Id']].tolist()
80
  output_dict = dict(zip(product_store_ids, predicted_prices)) # Use actual prices
81
 
82
  # Return the predictions dictionary as a JSON response
 
76
  predicted_prices = model.predict(input_data).tolist()
77
 
78
  # Create a dictionary of predictions with property IDs as keys
79
+ product_store_ids = input_data[['Product_Id', 'Store_Id']].values.tolist()
80
  output_dict = dict(zip(product_store_ids, predicted_prices)) # Use actual prices
81
 
82
  # Return the predictions dictionary as a JSON response