Upload folder using huggingface_hub
Browse files
app.py
CHANGED
|
@@ -48,6 +48,7 @@ def predict_product_batch():
|
|
| 48 |
|
| 49 |
# Read the file into a DataFrame
|
| 50 |
input_data = pd.read_csv(file)
|
|
|
|
| 51 |
|
| 52 |
# Make predictions for the batch data
|
| 53 |
predictions = model.predict(input_data).tolist()
|
|
|
|
| 48 |
|
| 49 |
# Read the file into a DataFrame
|
| 50 |
input_data = pd.read_csv(file)
|
| 51 |
+
input_data = input_data.drop(['Product_Id','Store_Establishment_Year','Product_Allocated_Area'],axis=1)
|
| 52 |
|
| 53 |
# Make predictions for the batch data
|
| 54 |
predictions = model.predict(input_data).tolist()
|