Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files
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 |
|