Update app.py

app.py

@@ -47,8 +47,8 @@ def segment_to_bbox(segment_indexs):
     return [np.min(x_points), np.min(y_points), np.max(x_points), np.max(y_points)]
 
 def clipseg_prediction(image):
-    img_w = image.width
-    img_h = image.height
+    
+    img_w, img_h = image.shape
     inputs = clip_processor(text=prompts, images=[image] * len(prompts), padding="max_length", return_tensors="pt")
     # predict
     with torch.no_grad():