app.py

app.py

@@ -9,6 +9,7 @@ import requests
 import gradio as gr
 import os
 from transformers import CLIPProcessor, CLIPModel, CLIPTokenizer
+import urllib.request
 
 # check if CUDA available 
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -34,9 +35,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).raw
+     #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)
      #photo = photo_id + '.jpg'
      #img = Image.open(response).convert("RGB")
      #img = Image.open(os.path.join(IMAGES_DIR, photo))
@@ -77,7 +79,7 @@ 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 = Image.fromarray(search_image.astype('uint8'), 'RGB')
+      #search_image = Image.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"]