Update app.py

app.py

@@ -48,7 +48,7 @@ def main():
         color:green;'> Classification pour l'octroi de credit </h2>", unsafe_allow_html = True)
     
     # charger le fichier
-    uploaded_file = st.sidebar.file_uploader("Upload your input CSV file", type=["csv"])
+    uploaded_file = st.sidebar.file_uploader("Importer fichier csv", type=["csv"])
     
     # creation du menu
     menu = ["Home", "Data Exploration", "Data Visualisation", "Make prediction"]
@@ -170,7 +170,10 @@ def main():
             st.write(data_encoded)
             
             # separation du jeu de donnee
-            X, y = data_encoded.drop(["Loan_Status_Y"], axis=1), data_encoded["Loan_Status_Y"]
+            if data_encoded.shape[1] == 15:
+                X, y = data_encoded.drop(["Loan_Status_Y"], axis=1), data_encoded["Loan_Status_Y"]
+            else:
+                X = data_encoded
             
             # traintement des valeurs manquantes
             sp = SimpleImputer(strategy="most_frequent")