Upload folder using huggingface_hub

app.py

@@ -63,7 +63,7 @@ def sales_predict():
   input_df_dummies = pd.get_dummies(input_data, columns=categorical_columns_for_dummies, drop_first=True)
 
   #make model to predict
-  prediction = model.predict(input_df_dummies.reindex(columns=X_train.columns, fill_value=0))
+  prediction = model.predict(input_df_dummies)
 
   return jsonify({'Prediction':prediction[0]})
 
@@ -83,7 +83,7 @@ def sales_batch_predict():
 #convert the categorical to dummies
   categorical_columns_for_dummies = ['Product_Sugar_Content','Product_Type','Store_Size','Store_Location_City_Type','Store_Type']
   input_df_dummies = pd.get_dummies(input_data, columns=categorical_columns_for_dummies, drop_first=True)
-  #input_df_aligned =input_df_dummies.reindex(columns=X_train.columns, fill_value=0)
+  
 
 #predict
   predictions = model.predict(input_df_aligned).tolist() # Predict and convert to list