Update app.py

app.py

@@ -146,9 +146,10 @@ def heatmap(top_n):
 
 
 
-def decode_plot(plot_base64):
+def decode_plot(plot_base64, top_n):
     plot_bytes = base64.b64decode(plot_base64['plot'].split(',')[1])
     img = plt.imread(BytesIO(plot_bytes), format='PNG')
+    plt.figure(figsize = (12, 2*top_n, dpi = 150)
     plt.imshow(img)
     plt.axis('off')
     plt.show()
@@ -165,7 +166,7 @@ def linePlot(viz_type, weight, top_n):
     )
     # print(result)
     # result is a tuble of dictionary of plot_base64, and a string message of description of the plot
-    return decode_plot(result[0]), result[1]
+    return decode_plot(result[0],top_n), result[1]