dawood HF staff commited on
Commit
98f80fc
Β·
1 Parent(s): b2bc265

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -1,5 +1,11 @@
1
  import gradio as gr
 
 
 
 
 
 
2
 
3
  gr.Interface.load(
4
- "spaces/abidlabs/remove-bg", inputs="webcam", title="Remove your webcam background!"
5
  ).launch()
 
1
  import gradio as gr
2
+ import torch
3
+
4
+ torch.hub.download_url_to_file('https://pbs.twimg.com/profile_images/691700243809718272/z7XZUARB_400x400.jpg',
5
+ 'demis.jpg')
6
+ torch.hub.download_url_to_file('https://hai.stanford.edu/sites/default/files/styles/person_medium/public/2020-03/hai_1512feifei.png?itok=INFuLABp',
7
+ 'lifeifei.png')
8
 
9
  gr.Interface.load(
10
+ "spaces/abidlabs/remove-bg", inputs="webcam", title="Remove your webcam background!", examples=[['demis.jpg'], ['lifeifei.png']]
11
  ).launch()