Update app.py

app.py

@@ -14,9 +14,12 @@ model_type = "vit_h"
 sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
 
 # Load the CLIP model
-model, preprocess = clip.load("ViT-B/32")
+# 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.to(device).eval()
+model, preprocess = clip.load("ViT-B/32", device=device)
 
 def find_similarity(base64_image, text_input):
     try: