jarpan03 commited on
Commit
b571bc7
·
verified ·
1 Parent(s): d8c0f01

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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()