Update app.py
Browse files
app.py
CHANGED
|
@@ -4,9 +4,7 @@ import torch
|
|
| 4 |
import numpy as np
|
| 5 |
from torchvision import transforms
|
| 6 |
|
| 7 |
-
|
| 8 |
-
description = "Automatically remove the image background from a profile photo."
|
| 9 |
-
article = "<p style='text-align: center'><a href='https://news.machinelearning.sg/posts/beautiful_profile_pics_remove_background_image_with_deeplabv3/'>Blog</a> | <a href='https://github.com/eugenesiow/practical-ml'>Github Repo</a></p>"
|
| 10 |
|
| 11 |
|
| 12 |
def make_transparent_foreground(pic, mask):
|
|
@@ -70,9 +68,8 @@ gr.Interface(
|
|
| 70 |
inference,
|
| 71 |
gr.inputs.Image(type="pil", label="Input"),
|
| 72 |
gr.outputs.Image(type="pil", label="Output"),
|
| 73 |
-
title=title,
|
| 74 |
description=description,
|
| 75 |
-
article=article,
|
| 76 |
examples=[['demis.jpg'], ['lifeifei.png']],
|
| 77 |
-
enable_queue=True
|
|
|
|
| 78 |
).launch(debug=False)
|
|
|
|
| 4 |
import numpy as np
|
| 5 |
from torchvision import transforms
|
| 6 |
|
| 7 |
+
description = "Automatically remove the image background from a profile photo. Based on a [Space by eugenesiow](https://huggingface.co/spaces/eugenesiow/remove-bg)."
|
|
|
|
|
|
|
| 8 |
|
| 9 |
|
| 10 |
def make_transparent_foreground(pic, mask):
|
|
|
|
| 68 |
inference,
|
| 69 |
gr.inputs.Image(type="pil", label="Input"),
|
| 70 |
gr.outputs.Image(type="pil", label="Output"),
|
|
|
|
| 71 |
description=description,
|
|
|
|
| 72 |
examples=[['demis.jpg'], ['lifeifei.png']],
|
| 73 |
+
enable_queue=True,
|
| 74 |
+
css=".footer{display:none !important}"
|
| 75 |
).launch(debug=False)
|