Update app.py

app.py

@@ -56,9 +56,9 @@ def do_process(img, baseline):
                          baselines=baselines,
                          target=predictions)
   attrs = explanation.attributions[0]
-  fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(20, 10))
+  fig, ax = plt.subplots(nrows=1, ncols=1, figsize=(10, 10))
   fig, ax = visualize_image_attr(attr=attrs.squeeze(), original_image=img, method='blended_heat_map',
-                      sign='all', show_colorbar=True, title='Overlaid Attributions',
+                      sign='all', show_colorbar=True, title=None,
                       plt_fig_axis=(fig, ax), use_pyplot=False)
   buf = io.BytesIO()
   fig.savefig(buf)