jarpan03 commited on
Commit
7bfeba2
·
verified ·
1 Parent(s): 1edd53d

Upload folder using huggingface_hub

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