ubamba98 commited on
Commit
54b0744
1 Parent(s): ff843a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,7 +1,9 @@
1
  import torch
 
2
  import numpy as np
3
  import pandas as pd
4
  import gradio as gr
 
5
  from PIL import Image as PILIMAGE
6
  from transformers import CLIPProcessor, CLIPModel
7
 
@@ -36,7 +38,8 @@ def find_similar(image):
36
  photo_id = photo_ids[idx]
37
  photo_data = photos[photos["photo_id"] == photo_id].iloc[0]
38
 
39
- return PILIMAGE(url=photo_data["photo_image_url"] + "?w=640")
40
-
 
41
 
42
  iface = gr.Interface(fn=find_similar, inputs=gr.inputs.Image(), outputs=gr.outputs.Image(type="pil")).launch()
 
1
  import torch
2
+ import requests
3
  import numpy as np
4
  import pandas as pd
5
  import gradio as gr
6
+ from io import BytesIO
7
  from PIL import Image as PILIMAGE
8
  from transformers import CLIPProcessor, CLIPModel
9
 
 
38
  photo_id = photo_ids[idx]
39
  photo_data = photos[photos["photo_id"] == photo_id].iloc[0]
40
 
41
+ response = requests.get(photo_data["photo_image_url"] + "?w=640")
42
+ img = PILIMAGE.open(BytesIO(response.content))
43
+ return img
44
 
45
  iface = gr.Interface(fn=find_similar, inputs=gr.inputs.Image(), outputs=gr.outputs.Image(type="pil")).launch()