Update app.py

app.py

@@ -95,6 +95,8 @@ def detect_objects2(model_name,url_input,image_input,threshold):
     
     #Visualize prediction
     viz_img = visualize_prediction(image, processed_outputs, threshold, model.config.id2label)
+
+    keep = processed_outputs["scores"] > threshold
     
     return processed_outputs["labels"][keep].tolist()