Update app.py

app.py

@@ -5,7 +5,6 @@ import joblib
 import pandas as pd
 import os
 
-
 def load_model():
     cwd = os.getcwd()
 
@@ -13,19 +12,16 @@ def load_model():
 
     Final_model_file_path = os.path.join(destination, "Final_model.joblib")
     preprocessor_file_path = os.path.join(destination, "preprocessor.joblib")
-    
 
     Final_model = joblib.load(Final_model_file_path)
     preprocessor = joblib.load(preprocessor_file_path)
-    
 
     return Final_model, preprocessor
 
 Final_model, preprocessor = load_model()
 
-
 #define prediction function
-def make_prediction(REGION, TENURE, MONTANT, FREQUENCE_RECH, REVENUE, ARPU_SEGMENT, FREQUENCE, DATA_VOLUME, ON_NET, ORANGE, TIGO, ZONE1, ZONE2,MRG, REGULARITY, FREQ_TOP_PACK):
+def make_prediction(REGION, TENURE, MONTANT, FREQUENCE_RECH, REVENUE, ARPU_SEGMENT, FREQUENCE, DATA_VOLUME, ON_NET, ORANGE, TIGO, ZONE1, ZONE2, MRG, REGULARITY, FREQ_TOP_PACK):
     #make a dataframe from input data
     input_data = pd.DataFrame({'REGION':[REGION], 
                                'TENURE':[TENURE], 
@@ -43,41 +39,50 @@ def make_prediction(REGION, TENURE, MONTANT, FREQUENCE_RECH, REVENUE, ARPU_SEGME
                                'MRG':[MRG],
                                'REGULARITY':[REGULARITY], 
                                'FREQ_TOP_PACK':[FREQ_TOP_PACK]})
-    
+
     transformer = preprocessor.transform(input_data)
-   
-    predt = Final_model.predict(transformer) 
+
+    predt = Final_model.predict(transformer)
     #return prediction
     if predt[0]==1:
         return "Customer will Churn" 
     return "Customer will not Churn"
-    
 
-#create the input components for gradio
-input_col1 = [ gr.inputs.Dropdown(choices =['DAKAR', 'THIES', 'SAINT-LOUIS', 'LOUGA', 'KAOLACK', 'DIOURBEL', 'TAMBACOUNDA' 'KAFFRINE,KOLDA', 'FATICK', 'MATAM', 'ZIGUINCHOR', 'SEDHIOU', 'KEDOUGOU']),
-               gr.inputs.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']), 
-               gr.inputs.Number(),
-               gr.Number(),
-               gr.Number(),
-               gr.Number(), 
-               gr.Number(), 
-               gr.Number()]
+# Create the input components for gradio
+input_col1 = gr.Column(
+    [
+        gr.Dropdown("REGION", choices=['DAKAR', 'THIES', 'SAINT-LOUIS', 'LOUGA', 'KAOLACK', 'DIOURBEL', 'TAMBACOUNDA', 'KAFFRINE', 'KOLDA']),
+        gr.Dropdown("TENURE", 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']),
+        gr.Number("MONTANT"),
+        gr.Number("FREQUENCE_RECH"),
+        gr.Number("REVENUE"),
+        gr.Number("ARPU_SEGMENT"),
+        gr.Number("FREQUENCE"),
+        gr.Number("DATA_VOLUME"),
+        gr.Number("ON_NET")
+    ],
+    label="Column 1"
+)
 
-input_col2 = [ gr.Number(),
-               gr.Number(),
-               gr.Number(),
-               gr.Number(),
-               gr.Number(),
-               gr.inputs.Dropdown(choices =['NO']),   
-               gr.Number(),
-               gr.Number()]
+input_col2 = gr.Column(
+    [
+        gr.Number("ORANGE"),
+        gr.Number("TIGO"),
+        gr.Number("ZONE1"),
+        gr.Number("ZONE2"),
+        gr.Dropdown("MRG", choices=['NO']),
+        gr.Number("REGULARITY"),
+        gr.Number("FREQ_TOP_PACK")
+    ],
+    label="Column 2"
+)
 
 output = gr.Textbox(label='Prediction')
-#create the interface component
 
-app = gr.Interface(fn =make_prediction,inputs=[input_col1, input_col2],
-                                                 title ="Customer Churn Predictor", 
-                                                  description="Enter the fields below and click the submit button to Make Your Prediction",
-                                                 outputs = output)
+# Create the interface component
+app = gr.Interface(fn=make_prediction, inputs=[input_col1, input_col2],
+                   title="Customer Churn Predictor",
+                   description="Enter the fields below and click the submit button to Make Your Prediction",
+                   outputs=output)
 
-app.launch(debug = True)
+app.launch(debug=True)