jwyang commited on
Commit
8424dda
1 Parent(s): 9d707d9
Files changed (1) hide show
  1. app.py +1 -1
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 Image.fromarray(heatmap), {texts.split(';')[i]: float(prediction[i]) for i in range(len(texts.split(';')))}
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()