setuagrawal commited on
Commit
c2e6df2
·
verified ·
1 Parent(s): 2fd7082

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -53,6 +53,10 @@ def predict_sales():
53
  # Convert to DataFrame
54
  input_data = pd.DataFrame(sales_data)
55
 
 
 
 
 
56
  # Predict
57
  prediction = model.predict(input_data).tolist()[0]
58
 
 
53
  # Convert to DataFrame
54
  input_data = pd.DataFrame(sales_data)
55
 
56
+ # Add Store_Id if not present (model expects it but doesn't use it)
57
+ if 'Store_Id' not in input_data.columns:
58
+ input_data['Store_Id'] = 'DUMMY_STORE'
59
+
60
  # Predict
61
  prediction = model.predict(input_data).tolist()[0]
62