Update app.py

app.py

@@ -18,11 +18,11 @@ def findCorrelation(dataset, target):
   print(df)
   print("\n")
   
-  d = df.corr()["coma"].to_dict()
+  d = df.corr()[target].to_dict()
   print(d)
   labels = sorted(d.items(), key=lambda x: x[1], reverse=True)
   
-  del labels[target]
+  d = d.pop(target)
   
   fig1 = plt.figure()
   hm = sns.heatmap(df.corr(), annot = True)
@@ -40,7 +40,7 @@ def findCorrelation(dataset, target):
   # use the function regplot to make a scatterplot
   sns.regplot(x=labels.keys()[2], y=df[target])
   
-  return labels, fig1, fig2, fig3, fig4
+  return d, fig1, fig2, fig3, fig4
 
 demo = gr.Interface(fn=findCorrelation, inputs=[gr.File(), 'text'], outputs=[gr.Label(), gr.Plot(), gr.Plot(), gr.Plot(), gr.Plot()], title="Find correlation")
 demo.launch(debug=True)