Spaces:
Runtime error
Runtime error
jaekookang
commited on
Commit
β’
806e6c5
1
Parent(s):
825e1d8
fix typo
Browse files
gradio_artist_classifier.py
CHANGED
@@ -5,6 +5,7 @@ prototype
|
|
5 |
---
|
6 |
- 2022-01-18 jkang first created
|
7 |
'''
|
|
|
8 |
from PIL import Image
|
9 |
import matplotlib.pyplot as plt
|
10 |
import matplotlib.image as mpimg
|
@@ -108,7 +109,7 @@ def predict(input_image):
|
|
108 |
fig.tight_layout()
|
109 |
|
110 |
buf = io.BytesIO()
|
111 |
-
fig.savefig(buf,
|
112 |
buf.seek(0)
|
113 |
pil_img = Image.open(buf)
|
114 |
plt.close()
|
@@ -132,4 +133,4 @@ iface = gr.Interface(
|
|
132 |
],
|
133 |
examples=EXAMPLES,
|
134 |
)
|
135 |
-
iface.launch(debug=True, enable_queue=True)
|
|
|
5 |
---
|
6 |
- 2022-01-18 jkang first created
|
7 |
'''
|
8 |
+
from gradcam_utils import get_img_4d_array, make_gradcam_heatmap, align_image_with_heatmap
|
9 |
from PIL import Image
|
10 |
import matplotlib.pyplot as plt
|
11 |
import matplotlib.image as mpimg
|
|
|
109 |
fig.tight_layout()
|
110 |
|
111 |
buf = io.BytesIO()
|
112 |
+
fig.savefig(buf, bbox_inches='tight', format='jpg')
|
113 |
buf.seek(0)
|
114 |
pil_img = Image.open(buf)
|
115 |
plt.close()
|
|
|
133 |
],
|
134 |
examples=EXAMPLES,
|
135 |
)
|
136 |
+
iface.launch(debug=True, enable_queue=True)
|