Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -52,10 +52,15 @@ class GOSNormalize(object):
|
|
| 52 |
|
| 53 |
|
| 54 |
transform = transforms.Compose([GOSNormalize([0.5,0.5,0.5],[1.0,1.0,1.0])])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 55 |
|
| 56 |
def load_image(im_path, hypar):
|
| 57 |
-
im_path = BytesIO(requests.get(im_path).content)
|
| 58 |
-
im =
|
| 59 |
im, im_shp = im_preprocess(im, hypar["cache_size"])
|
| 60 |
im = torch.divide(im,255.0)
|
| 61 |
shape = torch.from_numpy(np.array(im_shp))
|
|
@@ -155,7 +160,7 @@ article = "<div><center><img src='https://visitor-badge.glitch.me/badge?page_id=
|
|
| 155 |
|
| 156 |
interface = gr.Interface(
|
| 157 |
fn=inference,
|
| 158 |
-
inputs="
|
| 159 |
outputs=["image","image","image"],
|
| 160 |
title=title,
|
| 161 |
description=description,
|
|
|
|
| 52 |
|
| 53 |
|
| 54 |
transform = transforms.Compose([GOSNormalize([0.5,0.5,0.5],[1.0,1.0,1.0])])
|
| 55 |
+
def get_url_im(t):
|
| 56 |
+
user_agent = {'User-agent': 'gradio-app'}
|
| 57 |
+
response = requests.get(t, headers=user_agent)
|
| 58 |
+
return Image.open(BytesIO(response.content))
|
| 59 |
+
|
| 60 |
|
| 61 |
def load_image(im_path, hypar):
|
| 62 |
+
# im_path = BytesIO(requests.get(im_path).content)
|
| 63 |
+
im = get_url_im(im_path)
|
| 64 |
im, im_shp = im_preprocess(im, hypar["cache_size"])
|
| 65 |
im = torch.divide(im,255.0)
|
| 66 |
shape = torch.from_numpy(np.array(im_shp))
|
|
|
|
| 160 |
|
| 161 |
interface = gr.Interface(
|
| 162 |
fn=inference,
|
| 163 |
+
inputs=gr.Textbox(label="Text or Image URL", visible=False, interactive=True),
|
| 164 |
outputs=["image","image","image"],
|
| 165 |
title=title,
|
| 166 |
description=description,
|