Spaces:
Runtime error
Runtime error
fix input output
Browse files
app.py
CHANGED
@@ -6,6 +6,8 @@ from PIL import Image
|
|
6 |
from transformers import CLIPProcessor, CLIPModel
|
7 |
|
8 |
def find_similar(image):
|
|
|
|
|
9 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
10 |
|
11 |
## Define model
|
@@ -38,4 +40,4 @@ def find_similar(image):
|
|
38 |
return Image(url=photo_data["photo_image_url"] + "?w=640")
|
39 |
|
40 |
|
41 |
-
iface = gr.Interface(fn=find_similar, inputs=
|
|
|
6 |
from transformers import CLIPProcessor, CLIPModel
|
7 |
|
8 |
def find_similar(image):
|
9 |
+
|
10 |
+
image = Image.fromarray(image.astype('uint8'), 'RGB')
|
11 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
12 |
|
13 |
## Define model
|
|
|
40 |
return Image(url=photo_data["photo_image_url"] + "?w=640")
|
41 |
|
42 |
|
43 |
+
iface = gr.Interface(fn=find_similar, inputs=gr.inputs.Image(), outputs=gr.inputs.Image(type="pil").launch()
|