Update app.py

app.py

@@ -18,8 +18,6 @@ model, preprocess = clip.load("ViT-B/32")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device).eval()
 
-# device = "cuda" if torch.cuda.is_available() else "cpu"
-# model, preprocess = clip.load("ViT-B/32", device=device)
 
 def find_similarity(base64_image, text_input):
     try:
@@ -54,7 +52,7 @@ def segment_image(input_image, text_input):
     image_bytes = base64.b64decode(input_image)
     image = Image.open(BytesIO(image_bytes))
     
-    image = cv2.cvtColor(np.array(image), cv2.COLOR_BGR2RGB)
+    #image = cv2.cvtColor(np.array(image), cv2.COLOR_BGR2RGB)
     mask_generator = SamAutomaticMaskGenerator(sam)
     masks = mask_generator.generate(image)