Fix preprocessing bug

app.py

@@ -14,7 +14,6 @@ from segment_anything import SamAutomaticMaskGenerator, sam_model_registry
 CHECKPOINT_PATH = "sam_vit_h_4b8939.pth"
 MODEL_TYPE = "default"
 MAX_WIDTH = MAX_HEIGHT = 800
-CLIP_WIDTH = CLIP_HEIGHT = 300
 THRESHOLD = 0.05
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
@@ -85,8 +84,7 @@ def filter_masks(
  masked = image * np.expand_dims(mask["segmentation"], -1)
  crop = masked[y: y + h, x: x + w]
  crop = cv2.cvtColor(crop, cv2.COLOR_BGR2RGB)
- crop = PIL.Image.fromarray(crop * 255)
- crop.resize((CLIP_WIDTH, CLIP_HEIGHT))
+ crop = PIL.Image.fromarray(crop)
  cropped_masks.append(crop)
 
  if query and filtered_masks:
@@ -162,7 +160,7 @@ demo = gr.Interface(
  [
  0.9,
  0.8,
- 0.30,
+ 0.15,
  os.path.join(os.path.dirname(__file__), "examples/dog.jpg"),
  "A dog",
  ],
@@ -171,7 +169,7 @@ demo = gr.Interface(
  0.8,
  0.05,
  os.path.join(os.path.dirname(__file__), "examples/city.jpg"),
- "A bridge on the water",
+ "water",
  ],
  [
  0.9,