Update app.py

app.py

@@ -49,10 +49,8 @@ def findCorrelation(dataset, target):
   sns.regplot(x=df[keys[2][0]], y=df[target])
   
   print("\n Fig 4")
-  
-  labels = {key[0]: key[1] for key in keys[10]}
-  
-  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 = gr.Interface(fn=findCorrelation, inputs=[gr.File(), 'text'], outputs=[gr.Label(num_top_classes = 10), gr.Plot(), gr.Plot(), gr.Plot(), gr.Plot()], title="Find correlation")
 demo.launch(debug=True)