itacaiunas commited on
Commit
a0d7137
1 Parent(s): 375323c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -8
app.py CHANGED
@@ -5,17 +5,11 @@ title = "Remove Bg"
5
  description = "Automatically remove the image background from a profile photo."
6
  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>"
7
 
8
- torch.hub.download_url_to_file('https://pbs.twimg.com/profile_images/691700243809718272/z7XZUARB_400x400.jpg',
9
- 'demis.jpg')
10
- torch.hub.download_url_to_file('https://hai.stanford.edu/sites/default/files/styles/person_medium/public/2020-03/hai_1512feifei.png?itok=INFuLABp',
11
- 'lifeifei.png')
12
- model = torch.hub.load('pytorch/vision:v0.6.0', 'deeplabv3_resnet101', pretrained=True)
13
- model.eval()
14
 
15
  gr.Interface(
16
  inference,
17
- gr.inputs.Image(type="pil", label="Input"),
18
- gr.outputs.Image(type="pil", label="Output"),
19
  title=title,
20
  description=description,
21
  article=article,
 
5
  description = "Automatically remove the image background from a profile photo."
6
  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>"
7
 
8
+
 
 
 
 
 
9
 
10
  gr.Interface(
11
  inference,
12
+
 
13
  title=title,
14
  description=description,
15
  article=article,