Update app.py

app.py

@@ -12,7 +12,10 @@ def object_classify(img):
   # model predicts bounding boxes and corresponding COCO classes
   logits = outputs.logits
   bboxes = outputs.pred_boxes
-  return logits, bboxes
+  
+  probas = logits.softmax(-1)[0, :, :-1]
+  keep = probas.max(-1).values > 0.7
+  return keep
   
 interface=gr.Interface(fn=object_classify,
                        inputs=gr.inputs.Image(shape=(224,224),label='Insert Image'),