Update app.py

app.py

@@ -3,10 +3,9 @@ import numpy as np
 import gradio as gr
 import joblib
 import pandas as pd
-import itertools
 import os
 
-from gradio.components import Dropdown, Number, Textbox
+from gradio.components import Dropdown, Number, Textbox, Column
 
 def load_model():
     cwd = os.getcwd()
@@ -75,15 +74,13 @@ input_col2 = [Number(),
               Number(),
               Number()]
 
-def flatten(list_of_lists):
-    return itertools.chain.from_iterable(list_of_lists)
-
-inputs = flatten([input_col1, input_col2])
-
 output = Textbox(label='Prediction')
 
+# Wrap the input components in a Column component
+inputs = Column(children=[input_col1, input_col2])
+
 # Create the interface component
-app = gr.Interface(fn=make_prediction, inputs=[input_col1, input_col2],
+app = gr.Interface(fn=make_prediction, inputs=inputs,
                    title="Customer Churn Predictor",
                    description="Enter the fields below and click the submit button to Make Your Prediction",
                    outputs=output)