Update app.py

app.py

@@ -46,12 +46,12 @@ def detect_objects(model_name, image_input, threshold, display_mask=False, img_i
         mask_pil_image = img_input_mask
 
     viz_img = visualize_prediction(
-        image_input,
-        processed_outputs,
-        threshold,
-        model.config.id2label,
-        display_mask,
-        mask_pil_image
+        pil_img=image_input,
+        output_dict=processed_outputs,
+        threshold=threshold,
+        id2label=model.config.id2label,
+        display_mask=display_mask,
+        mask=mask_pil_image
     )
     decoder_attention_map_img = visualize_attention_map(
         image_input, decoder_attention_map