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,