Update return to fig

app.py

@@ -167,14 +167,12 @@ def gradio_music_graph(client_id, client_secret, artist_name): #total_albums
   plt.tight_layout()
 
   sns.boxplot(x=df["album_name"], y=df["popularity"], ax=ax)
-  fig.savefig('artist_popular_albums.png')
+  #fig.savefig('artist_popular_albums.png')
   plt.show()
-  # plt.plot(projected_values.T)
-  # plt.legend(employee_data["Name"])
-    # return employee_data, plt.gcf(), regression_values
-  return df
+  
+  return fig
 #Interface will include these buttons based on parameters in the function with a dataframe output
 music_plots = gr.Interface(gradio_music_graph, ["text", "text", "text"], 
-                           ["dataframe"], title="Popular Songs By Album Box Plot Distribution on Spotify", description="Using your Spotify API Access from https://developer.spotify.com/ you can see your favourite artist's most popular albums on Spotify")
+                           ["plot"], title="Popular Songs By Album Box Plot Distribution on Spotify", description="Using your Spotify API Access from https://developer.spotify.com/ you can see your favourite artist's most popular albums on Spotify")
 
 music_plots.launch(debug=True)