Update app.py

app.py

@@ -24,16 +24,27 @@ def findCorrelation(dataset, target):
   hm = sns.heatmap(df.corr(), annot = True)
   hm.set(title = "Correlation matrix of dataset\n")
     
-  fig2 = plt.figure()
-  sns.regplot(x=df[keys[0][0]], y=df[target])
+  try:
+    fig2 = plt.figure()
+    sns.regplot(x=df[keys[0][0]], y=df[target])
+  except:
+    fig2 = plt.figure()
   
-  fig3 = plt.figure()
-  sns.regplot(x=df[keys[1][0]], y=df[target])
+  try:
+    fig3 = plt.figure()
+    sns.regplot(x=df[keys[1][0]], y=df[target])
+  except:
+    fig3 = plt.figure()
 
-  fig4 = plt.figure()
-  sns.regplot(x=df[keys[2][0]], y=df[target])
-   
+  try:
+    fig4 = plt.figure()
+    sns.regplot(x=df[keys[2][0]], y=df[target])
+  except:
+    fig4 = plt.figure()
+    
   return d, fig1, fig2, fig3, fig4
 
+examples = [["boston.csv", "MEDV"]]
+
 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)