app.py

app.py

@@ -17,8 +17,6 @@ model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
 processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
 tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-base-patch32")
 
-model = model.to(device)
-
 # taking photo IDs
 photo_ids = pd.read_csv("./photo_ids.csv")
 photo_ids = list(photo_ids['photo_id'])
@@ -44,7 +42,7 @@ def encode_search_query(search_query, model):
         #inputs = tokenizer([search_query],  padding=True, return_tensors="pt")
         inputs = processor(text=[search_query], images=None, return_tensors="pt", padding=True)
     text_features =  model.get_text_features(**inputs).detach().numpy()
-    return np.array(text_features)
+    return text_features
         
 # Find all matched photos
 def find_matches(text_features, photo_features, photo_ids, results_count=4):