Update app.py

app.py

@@ -46,6 +46,9 @@ if st.button("Predict"):
     numerical_cols = ['day', 'month', 'year', 'shop_id', 'item_id', 'item_price', 'item_category_id']
     input_df = pd.DataFrame(input_data, index=[0])[numerical_cols]
     
+    # Fit the model
+    rf_model.fit(input_df)  # Fit the model with the scaled input data
+    
     # Make predictions using the trained model
     predictions = rf_model.predict(input_df)