Update app.py

app.py

@@ -9,17 +9,20 @@ def findCorrelation():
   df = pd.DataFrame(data=iris.data, columns=iris.feature_names)
 
   df["target"] = iris.target
-
+  
   correlation = df["sepal length (cm)"].corr(df["petal length (cm)"])
   corr = df.corr()
-
+  
+  fig1 = plt.figure()
   hm = sns.heatmap(df.corr(), annot = True)
   hm.set(xlabel='\nIRIS Flower Details', ylabel='IRIS Flower Details\t', title = "Correlation matrix of IRIS data\n")
-
+  
+  fig2 = plt.figure()
   # use the function regplot to make a scatterplot
   sns.regplot(x=df["sepal length (cm)"], y=df["sepal width (cm)"])
   plt.show()
-  return plt, plt
+  
+  return fig1, fig2, plt
 
-demo = gr.Interface(fn=findCorrelation, inputs=[], outputs=[gr.Plot(), gr.Plot()], title="Find correlation")
+demo = gr.Interface(fn=findCorrelation, inputs=[], outputs=[gr.Plot(), gr.Plot(), gr.Plot()], title="Find correlation")
 demo.launch()