Update app.py

app.py

@@ -42,16 +42,13 @@ def detect_objects(model_name, image_input, threshold, display_mask=False, img_i
         cross_attention_map,
     ) = make_prediction(image_input, feature_extractor, model)
 
-    if display_mask:
-        mask_pil_image = img_input_mask
-
     viz_img = visualize_prediction(
         pil_img=image_input,
         output_dict=processed_outputs,
         threshold=threshold,
         id2label=model.config.id2label,
         display_mask=display_mask,
-        mask=mask_pil_image
+        mask=img_input_mask
     )
     decoder_attention_map_img = visualize_attention_map(
         image_input, decoder_attention_map