Update app.py

app.py

@@ -14,8 +14,9 @@ model_type = "vit_h"
 sam = sam_model_registry[model_type](checkpoint=sam_checkpoint)
 
 # Load the CLIP model
+model, preprocess = clip.load("ViT-B/32")
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model, preprocess = clip.load("ViT-B/32", device=device)
+model.to(device).eval()
 
 def find_similarity(base64_image, text_input):
     try: