app.py

app.py

@@ -36,7 +36,7 @@ def show_output_image(matched_images) :
      #photo_image_url = f"https://unsplash.com/photos/{photo_id}?ixid=2yJhcHBfaWQiOjEyMDd9&fm=jpg"
      #photo_found = photos[photos["photo_id"] == photo_id].iloc[0]
      #response = requests.get(photo_found["photo_image_url"] + "?w=640")
-     response = requests.get(photo_image_url, stream=True).raw
+     response = requests.get(photo_image_url, stream=True)
      img = Image.open(BytesIO(response.content))
      #return img
      #photo_jpg = photo_id + '.jpg'
@@ -52,9 +52,11 @@ def encode_search_query(search_query, model, device):
         #text_encoded = model.encode_text(clip.tokenize(search_query).to(device))
         #text_encoded /= text_encoded.norm(dim=-1, keepdim=True)
         # Retrieve the feature vector from the GPU and convert it to a numpy array
-    text_features = model.get_text_features(**inputs).detach().numpy()
+    #text_features = model.get_text_features(**inputs).detach().numpy()
     #text_features = model.get_text_features(**inputs).cpu().numpy()
-    return text_features
+    text_features =  model.get_text_features(**inputs).detach().numpy()
+    return np.array(text_features)
+    #return text_features
         #return text_encoded.cpu().numpy()
         
 # Find all matched photos
@@ -79,13 +81,7 @@ def image_search(search_text, search_image, option):
   
     # Find the matched Images
     matched_images = find_matches(text_features, photo_features, photo_ids, 4)
-    # ---- debug purpose ------#
-    #print(matched_images[0])
-    #id = matched_images[0]
-    #photo_image_url = f"https://unsplash.com/photos/{id}/download?w=280"
-    #print(photo_image_url)
-    #--------------------------#
-  
+    
     return show_output_image(matched_images)
   elif option == "Image-To-Image":
       # Input Image for Search