ubamba98 commited on
Commit
b89ace4
·
1 Parent(s): 598e977

fix input output

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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="image", outputs="image").launch()
 
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()