Update app.py

app.py

@@ -11,9 +11,9 @@ def findCorrelation(dataset, target):
   
   non_numeric_cols = df.select_dtypes('object').columns.tolist()
   
-  for non_numeric_col in non_numeric_cols:
+  if df[target] in non_numeric_cols:
     label_encoder = LabelEncoder()
-    df[non_numeric_col] = label_encoder.fit_transform(df[non_numeric_col])
+    df[non_numeric_col] = label_encoder.fit_transform(df[target])
   
   d = df.corr()[target].to_dict()
   d.pop(target)