Update app.py

app.py

@@ -32,22 +32,26 @@ def process_image(image, prompt, threshold, alpha_value, draw_rectangles):
 
     mask = Image.fromarray(np.uint8(mat * 255), "L")  # Convert to PIL Image
 
+    # Convert the mask to a NumPy array for calculation
+    mask_array = np.array(mask)
+
     # normalize the mask
-    mask_min = mask.min()
-    mask_max = mask.max()
-    mask = (mask - mask_min) / (mask_max - mask_min)
+    mask_min = mask_array.min()
+    mask_max = mask_array.max()
+    mask_array = (mask_array - mask_min) / (mask_max - mask_min)
 
     # threshold the mask
-    bmask = mask > threshold
+    bmask = mask_array > threshold
     # zero out values below the threshold
-    mask[mask < threshold] = 0
+    mask_array[mask_array < threshold] = 0
 
-    bmask = Image.fromarray(bmask.astype(np.uint8) * 255, "L")
+    bmask = Image.fromarray((bmask * 255).astype(np.uint8), "L")
 
     return bmask
 
 
 
+
 @app.route('/')
 def index():
     return "Hello, World! clipseg2"