Update app.py

app.py

@@ -5,6 +5,8 @@ import joblib
 import pandas as pd
 import os
 
+from gradio.components import Dropdown, Number, Textbox
+
 def load_model():
     cwd = os.getcwd()
 
@@ -49,29 +51,30 @@ def make_prediction(REGION, TENURE, MONTANT, FREQUENCE_RECH, REVENUE, ARPU_SEGME
     return "Customer will not Churn"
 
 # Create the input components for gradio
-input_col1 = [
-    gr.Dropdown(['DAKAR', 'THIES', 'SAINT-LOUIS', 'LOUGA', 'KAOLACK', 'DIOURBEL', 'TAMBACOUNDA', 'KAFFRINE', 'KOLDA']),
-    gr.Dropdown(['K > 24 month', 'I 18-21 month', 'H 15-18 month', 'G 12-15 month', 'J 21-24 month', 'F 9-12 month', 'E 6-9 month', 'D 3-6 month']),
-    gr.Number(),
-    gr.Number(),
-    gr.Number(),
-    gr.Number(),
-    gr.Number(),
-    gr.Number(),
-    gr.Number()
-]
+input_col1 = [Dropdown(choices=['DAKAR', 'THIES', 'SAINT-LOUIS', 'LOUGA', 'KAOLACK', 'DIOURBEL', 'TAMBACOUNDA', 'KAFFRINE', 'KOLDA']),
+              Dropdown(choices=['K > 24 month', 'I 18-21 month', 'H 15-18 month', 'G 12-15 month', 'J 21-24 month', 'F 9-12 month', 'E 6-9 month', 'D 3-6 month']),
+              Number(),
+              Number(),
+              Number(),
+              Number(),
+              Number(),
+              Number(),
+              Number()]
 
-input_col2 = [
-    gr.Number(),
-    gr.Number(),
-    gr.Number(),
-    gr.Number(),
-    gr.Dropdown(['NO']),
-    gr.Number(),
-    gr.Number()
-]
+input_col2 = [Number(),
+              Number(),
+              Number(),
+              Number(),
+              Number(),
+              Number(),
+              Number(),
+              Number(),
+              Number(),
+              Dropdown(choices=['NO']),
+              Number(),
+              Number()]
 
-output = gr.Textbox(label='Prediction')
+output = Textbox(label='Prediction')
 
 # Create the interface component
 app = gr.Interface(fn=make_prediction, inputs=[input_col1, input_col2],