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