Update train/train.py

train/train.py

@@ -1,9 +1,16 @@
 import pandas as pd
-import lightgbm as lgb
 from sklearn.model_selection import train_test_split
 from sklearn.preprocessing import LabelEncoder
 from imblearn.over_sampling import SMOTE
 
+try:
+    import lightgbm as lgb
+except:
+    print("An exception occurred")
+    !pip install lightgbm
+    import lightgbm as lgb
+
+    
 # Load the dataset
 file_path = 'datasets'  # Update with your dataset path
 df = pd.read_excel(file_path + '/dataset.xlsx')