Add print statement to display predicted masks in the prediction function

main.py

@@ -76,6 +76,7 @@ def predict(image, prompts):
 
     # Get predictions
     pred_mask = interactive_infer_image(model, image, prompts)
+    print(f"Predicted masks: {pred_mask}")
 
     # Generate visualization
     colors = generate_colors(len(prompts))