Spaces:
Build error
Build error
jwyang
commited on
Commit
·
8424dda
1
Parent(s):
9d707d9
fix bug
Browse files
app.py
CHANGED
@@ -131,7 +131,7 @@ def recognize_image(image, texts):
|
|
131 |
output_map = (output_map - output_map.min()) / (output_map.max() - output_map.min())
|
132 |
heatmap = show_cam_on_image(img_d, output_map, use_rgb=True)
|
133 |
|
134 |
-
return
|
135 |
|
136 |
|
137 |
image = gr.inputs.Image()
|
|
|
131 |
output_map = (output_map - output_map.min()) / (output_map.max() - output_map.min())
|
132 |
heatmap = show_cam_on_image(img_d, output_map, use_rgb=True)
|
133 |
|
134 |
+
return {texts.split(';')[i]: float(prediction[i]) for i in range(len(texts.split(';')))}, Image.fromarray(heatmap)
|
135 |
|
136 |
|
137 |
image = gr.inputs.Image()
|