fix: adding debug prints

app.py

@@ -76,9 +76,10 @@ def interface(classifier):
             image_np = image
 
         mask = results["segmentation_mask"]
+        print(mask)
 
         overlay = image_np.copy()
-        overlay[mask < 0.6] = overlay[mask < 0.6] * 0
+        overlay[mask < 0.2] = overlay[mask < 0.2] * 0
 
         output_str = f"Predicted Class: {results['predicted_class']}\n"
         output_str += f"Confidence: {results['confidence']*100:.2f}%\n\n"