fix: double norm

app.py

@@ -87,8 +87,7 @@ def interface(classifier):
         # Use the normalized mask for visualization
         mask_viz = (normalized_mask * 255).astype(np.uint8)
 
-
-        binary_for_overlay = normalized_mask > 0.5
+        binary_for_overlay = normalized_mask > 0.2
         overlay = image_np.copy()
         overlay[~binary_for_overlay] = (overlay[~binary_for_overlay] * 0.3).astype(np.uint8)
 
@@ -103,8 +102,6 @@ def interface(classifier):
         for class_name, prob in sorted_probs:
             output_str += f"{class_name}: {prob*100:.2f}%\n"
 
-        mask_viz = (mask * 255).astype(np.uint8)
-
         return [output_str, overlay, mask_viz]
 
     demo = gr.Interface(