Update app.py

app.py

@@ -1,7 +1,27 @@
 import gradio as gr
+from torchvision import transforms
 
-titulo = "Remoção de Fundo de Imagem usando Gradio"
-descricao = "Este código usa a biblioteca Gradio para criar uma interface que permite remover o fundo de uma imagem. Ele carrega um modelo treinado previamente e permite que os usuários carreguem suas próprias imagens para remoção do fundo. O modelo usado é o 'spaces/eugenesiow/remove-bg'. Experimente agora!"
+title = "Remove Bg"
+description = "Automatically remove the image background from a profile photo."
+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>"
+
+torch.hub.download_url_to_file('https://pbs.twimg.com/profile_images/691700243809718272/z7XZUARB_400x400.jpg',
+                               'demis.jpg')
+torch.hub.download_url_to_file('https://hai.stanford.edu/sites/default/files/styles/person_medium/public/2020-03/hai_1512feifei.png?itok=INFuLABp',
+                               'lifeifei.png')
+model = torch.hub.load('pytorch/vision:v0.6.0', 'deeplabv3_resnet101', pretrained=True)
+model.eval()
+
+gr.Interface(
+    inference,
+    gr.inputs.Image(type="pil", label="Input"),
+    gr.outputs.Image(type="pil", label="Output"),
+    title=title,
+    description=description,
+    article=article,
+    examples=[['demis.jpg'], ['lifeifei.png']],
+    enable_queue=True
+).launch(debug=False)
 
 gr.Interface.load("spaces/eugenesiow/remove-bg").launch()