Update app.py

app.py

@@ -35,23 +35,22 @@ def findCorrelation(dataset, target):
   fig2 = plt.figure()
   # use the function regplot to make a scatterplot
   print(keys[0])
-  sns.regplot(x=df[keys[0]], y=df[target])
-  print(df[keys[0]])  
+  sns.regplot(x=df[keys[0][0]], y=df[target])
   print("\n Fig 2")
   
   fig3 = plt.figure()
   # use the function regplot to make a scatterplot
-  sns.regplot(x=df[keys[1]], y=df[target])
+  sns.regplot(x=df[keys[1][0]], y=df[target])
 
   print("\n Fig 3")
 
   fig4 = plt.figure()
   # use the function regplot to make a scatterplot
-  sns.regplot(x=df[keys[2]], y=df[target])
+  sns.regplot(x=df[keys[2][0]], y=df[target])
   
   print("\n Fig 4")
   
-  labels = {key: d[key] for key in keys[10]}
+  labels = {key: value for key, value in keys[10]}
   
   return labels, fig1, fig2, fig3, fig4