Updated app

app.py

@@ -32,7 +32,7 @@ def get_output_figure(pil_img, scores, labels, boxes):
 
     return plt.gcf()
 
-def get_output_attn_figure(image, encoding, results):
+def get_output_attn_figure(image, encoding, results, outputs):
     # keep only predictions of queries with +0.9 condifence (excluding no-object class)
     probas = outputs.logits.softmax(-1)[0, :, :-1]
     keep = probas.max(-1).values > 0.9
@@ -103,7 +103,7 @@ def detect(image):
     buf.seek(0)
     output_pil_img = Image.open(buf)
 
-    output_figure_attn = get_output_attn_figure(image, encoding, results)
+    output_figure_attn = get_output_attn_figure(image, encoding, results, outputs)
 
     buf = io.BytesIO()
     output_figure_attn.savefig(buf, bbox_inches='tight')