Update app.py

app.py

@@ -96,7 +96,7 @@ def detect_objects2(model_name,url_input,image_input,threshold):
     #Visualize prediction
     viz_img = visualize_prediction(image, processed_outputs, threshold, model.config.id2label)
     
-    return processed_outputs 
+    return processed_outputs["labels"][keep].tolist() 
     
 def set_example_image(example: list) -> dict:
     return gr.Image.update(value=example[0])