Spaces:

sergiopaniego
/

DETR_object_detection

Sleeping

sergiopaniego commited on Jun 18

Commit

13fc488

•

1 Parent(s): 153ddd7

Added attention

Files changed (1) hide show

app.py CHANGED Viewed

@@ -119,6 +119,6 @@ demo = gr.Interface(
     inputs=gr.Image(label="Input image", type="pil"),
     outputs=[
         gr.Image(label="Output image predictions", type="pil"),
-        gr.Image(label="Output image predictions", type="pil")
         ])
 demo.launch()

     inputs=gr.Image(label="Input image", type="pil"),
     outputs=[
         gr.Image(label="Output image predictions", type="pil"),
+        gr.Image(label="Output attention weights", type="pil")
         ])
 demo.launch()