Update app.py

app.py

@@ -56,7 +56,7 @@ with open('./trained_model.pkl', 'rb') as file:
 new_pred = st.text_area('Enter text')
 
 if new_pred:
-    new_pred = {key:0 for key in X_train.columns}
+    new_pred = {key:0 for key in columns}
     new_pred['date'] = pd.to_datetime('2014-07-10') # do not change
     
     new_pred['bedrooms'] = 5