RobotJelly commited on
Commit
ffaf3d7
·
1 Parent(s): 60b18f5
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -9,6 +9,7 @@ import requests
9
  import gradio as gr
10
  import os
11
  from transformers import CLIPProcessor, CLIPModel, CLIPTokenizer
 
12
 
13
  # check if CUDA available
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -34,9 +35,10 @@ def show_output_image(matched_images) :
34
  image=[]
35
  for photo_id in matched_images:
36
  photo_image_url = f"https://unsplash.com/photos/{photo_id}/download?w=280"
37
- response = requests.get(photo_image_url, stream=True)
38
- img = Image.open(BytesIO(response.content))
39
- #response = requests.get(photo_image_url, stream=True).raw
 
40
  #photo = photo_id + '.jpg'
41
  #img = Image.open(response).convert("RGB")
42
  #img = Image.open(os.path.join(IMAGES_DIR, photo))
@@ -77,7 +79,7 @@ def image_search(search_text, search_image, option):
77
  return show_output_image(matched_images)
78
  elif option == "Image-To-Image":
79
  # Input Image for Search
80
- search_image = Image.fromarray(search_image.astype('uint8'), 'RGB')
81
 
82
  with torch.no_grad():
83
  processed_image = processor(text=None, images=search_image, return_tensors="pt", padding=True)["pixel_values"]
 
9
  import gradio as gr
10
  import os
11
  from transformers import CLIPProcessor, CLIPModel, CLIPTokenizer
12
+ import urllib.request
13
 
14
  # check if CUDA available
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
35
  image=[]
36
  for photo_id in matched_images:
37
  photo_image_url = f"https://unsplash.com/photos/{photo_id}/download?w=280"
38
+ #response = requests.get(photo_image_url, stream=True)
39
+ #img = Image.open(BytesIO(response.content))
40
+ response = requests.get(photo_image_url, stream=True).raw
41
+ img = Image.open(response)
42
  #photo = photo_id + '.jpg'
43
  #img = Image.open(response).convert("RGB")
44
  #img = Image.open(os.path.join(IMAGES_DIR, photo))
 
79
  return show_output_image(matched_images)
80
  elif option == "Image-To-Image":
81
  # Input Image for Search
82
+ #search_image = Image.fromarray(search_image.astype('uint8'), 'RGB')
83
 
84
  with torch.no_grad():
85
  processed_image = processor(text=None, images=search_image, return_tensors="pt", padding=True)["pixel_values"]