Spaces:

CVPR
/

unicl-zero-shot-img-recog

Build error

jwyang commited on Jun 15, 2022

Commit

3c4c6c8

1 Parent(s): eb1d5d5

swap output

Files changed (1) hide show

app.py CHANGED Viewed

@@ -131,7 +131,7 @@ def recognize_image(image, texts):
     output_map = (output_map - output_map.min()) / (output_map.max() - output_map.min())
     heatmap = show_cam_on_image(img_d, output_map, use_rgb=True)
-    return {texts.split(';')[i]: float(prediction[i]) for i in range(len(texts.split(';')))}, Image.fromarray(heatmap)
 image = gr.inputs.Image()

     output_map = (output_map - output_map.min()) / (output_map.max() - output_map.min())
     heatmap = show_cam_on_image(img_d, output_map, use_rgb=True)
+    return Image.fromarray(heatmap), {texts.split(';')[i]: float(prediction[i]) for i in range(len(texts.split(';')))}
 image = gr.inputs.Image()