Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -60,6 +60,7 @@ def do_process(img, baseline):
|
|
60 |
fig, ax = visualize_image_attr(attr=attrs.squeeze(), original_image=img, method='blended_heat_map',
|
61 |
sign='all', show_colorbar=True, title=None,
|
62 |
plt_fig_axis=(fig, ax), use_pyplot=False)
|
|
|
63 |
buf = io.BytesIO()
|
64 |
fig.savefig(buf)
|
65 |
buf.seek(0)
|
|
|
60 |
fig, ax = visualize_image_attr(attr=attrs.squeeze(), original_image=img, method='blended_heat_map',
|
61 |
sign='all', show_colorbar=True, title=None,
|
62 |
plt_fig_axis=(fig, ax), use_pyplot=False)
|
63 |
+
fig.tight_layout()
|
64 |
buf = io.BytesIO()
|
65 |
fig.savefig(buf)
|
66 |
buf.seek(0)
|