Update app.py

app.py

@@ -25,9 +25,7 @@ def process_image(image, prompt, threshold, alpha_value, draw_rectangles):
 
     pred = torch.sigmoid(preds)
     mat = pred.cpu().numpy()
-    mask = Image.fromarray(np.uint8(mat * 255), "L")
-    mask = mask.resize(image.size)
-    mask = np.array(mask)[:, :, 0]
+    mask = Image.fromarray(np.uint8(mat[0, 0] * 255), "L")  # Access the first channel of the output
 
     # normalize the mask
     mask_min = mask.min()
@@ -43,6 +41,7 @@ def process_image(image, prompt, threshold, alpha_value, draw_rectangles):
 
     return bmask
 
+
 @app.route('/')
 def index():
     return "Hello, World! clipseg2"