Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
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()
|