Cacau commited on
Commit
efcf8a5
1 Parent(s): 7a4181f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -0
app.py CHANGED
@@ -2,6 +2,19 @@ import gradio as gr
2
  import requests
3
  import io
4
  from PIL import Image
 
 
 
 
 
 
 
 
 
 
 
 
 
5
 
6
  def greet(name):
7
  return "Hello " + name + "!!"
 
2
  import requests
3
  import io
4
  from PIL import Image
5
+ from transformers import pipeline
6
+ from torchvision import transforms
7
+
8
+ title = "Fine Tuned SD Model - Authoral stylization"
9
+ description = "Automatically remove the image background from a profile photo."
10
+ 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>"
11
+
12
+
13
+ gr.Interface.load(
14
+ "spaces/eugenesiow/remove-bg",
15
+
16
+ inputs=[gr.Image(label="Input Image", source="webcam")]
17
+ ).launch()
18
 
19
  def greet(name):
20
  return "Hello " + name + "!!"