Update app.py

app.py

@@ -137,12 +137,13 @@ if menu == 'Prediction target':
     train_enc = input_data.drop(['city'],axis = 1)
     #input_d = pd.concat([train_enc, encoded_cat], axis=1)
     input_d = pd.concat([train_enc, input_encoded_df], axis=1)
+    prediction = input_d.values
 
 
 
     # convert input_data to a numpy array before flattening to convert it back to a 2D array
     input_df= input_d.to_numpy()
-    prediction = model.predict(input_df.flatten().reshape(1, -1))
+    prediction = model.predict(prediction.flatten().reshape(1, -1))
     
 
     if st.button('Predict'):