Update app.py

app.py

@@ -14,12 +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")
-# device = "cuda" if torch.cuda.is_available() else "cpu"
-# model.to(device).eval()
-
+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()
+
+# 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:
@@ -82,7 +82,7 @@ def segment_image(input_image, text_input):
 
 # Create Gradio components
 input_image = gr.inputs.Image()
-text_input = gr.inputs.Textbox(label="Text Input")  # Use Textbox instead of Text
+text_input = gr.inputs.Textbox(label= "Text Input")  # Use Textbox instead of Text
 output_images = gr.outputs.JSON()
 
 # Create a Gradio interface