Spaces:
Build error
Build error
Commit
·
ffaf3d7
1
Parent(s):
60b18f5
app.py
Browse files
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 |
-
|
|
|
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"]
|