app.py

app.py

@@ -42,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 text_features
+    return np.array(text_features)
         
 # Find all matched photos
 def find_matches(text_features, photo_features, photo_ids, results_count=4):
@@ -51,7 +51,8 @@ def find_matches(text_features, photo_features, photo_ids, results_count=4):
   # Sort the photos by their similarity score
   best_photo_idx = (-similarities).argsort()
   # Return the photo IDs of the best matches
-  return [photo_ids[i] for i in best_photo_idx[:results_count]]
+  matches = [photo_ids[i] for i in best_photo_idx[:results_count]]
+  return np.array(matches)
   
 def image_search(search_text, search_image, option):