app.py

app.py

@@ -31,8 +31,10 @@ def show_output_image(matched_images) :
    image=[]
    for photo_id in matched_images:
      photo_image_url = f"https://unsplash.com/photos/{photo_id}/download?w=280"
-     response = requests.get(photo_image_url, stream=True)
-     img = Image.open(BytesIO(response.content))
+     #response = requests.get(photo_image_url, stream=True)
+     #img = Image.open(BytesIO(response.content))
+     response = requests.get(photo_image_url, stream=True).raw
+     img = Image.open(response).convert("RGB")
      image.append(img)
    return image
    
@@ -42,7 +44,7 @@ def encode_search_query(search_query, model, device):
         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):
@@ -70,6 +72,8 @@ def image_search(search_text, search_image, option):
     return show_output_image(matched_images)
   elif option == "Image-To-Image":
       # Input Image for Search
+      search_image = PILIMAGE.fromarray(search_image.astype('uint8'), 'RGB')
+      
       with torch.no_grad():
          processed_image = processor(text=None, images=search_image, return_tensors="pt", padding=True)["pixel_values"]
          image_feature = model.get_image_features(processed_image.to(device))