HybaAI commited on
Commit
1c01319
·
verified ·
1 Parent(s): 0fa351c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -11
app.py CHANGED
@@ -7,19 +7,12 @@ from briarmbg import BriaRMBG
7
  import PIL
8
  from PIL import Image
9
  from typing import Tuple
10
- from googletrans import Translator
11
 
12
  # Carregar o modelo pré-treinado
13
  net = BriaRMBG.from_pretrained("briaai/RMBG-1.4")
14
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
15
  net.to(device)
16
 
17
- # Inicializar o tradutor
18
- translator = Translator()
19
-
20
- def traduzir_texto(texto):
21
- return translator.translate(texto, dest='pt').text
22
-
23
  # Função para redimensionar a imagem para o tamanho que o modelo espera
24
  def redimensionar_imagem(imagem):
25
  imagem = imagem.convert('RGB')
@@ -62,21 +55,21 @@ def processar(imagem):
62
  return nova_im
63
 
64
  # Interface com Gradio
65
- gr.Markdown("") # Você pode adicionar um texto aqui, se desejar
66
 
67
  # Exemplos
68
  exemplos = [['./input.jpg']]
69
 
70
  # Configurando a interface
71
- output = gr.Image(type="pil", label=traduzir_texto("Imagem Processada"))
72
 
73
  # Definindo a interface com inputs e outputs
74
  demo = gr.Interface(
75
  fn=processar,
76
- inputs=gr.Image(type="numpy", label=traduzir_texto("Carregar Imagem")),
77
  outputs=output,
78
  examples=exemplos
79
  )
80
 
81
  # Executando a interface
82
- demo.launch(share=False)
 
7
  import PIL
8
  from PIL import Image
9
  from typing import Tuple
 
10
 
11
  # Carregar o modelo pré-treinado
12
  net = BriaRMBG.from_pretrained("briaai/RMBG-1.4")
13
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
14
  net.to(device)
15
 
 
 
 
 
 
 
16
  # Função para redimensionar a imagem para o tamanho que o modelo espera
17
  def redimensionar_imagem(imagem):
18
  imagem = imagem.convert('RGB')
 
55
  return nova_im
56
 
57
  # Interface com Gradio
58
+ gr.Markdown("")
59
 
60
  # Exemplos
61
  exemplos = [['./input.jpg']]
62
 
63
  # Configurando a interface
64
+ output = gr.Image(type="pil", label="Imagem Processada")
65
 
66
  # Definindo a interface com inputs e outputs
67
  demo = gr.Interface(
68
  fn=processar,
69
+ inputs=gr.Image(type="numpy", label="Carregar Imagem"), # Mantido em português
70
  outputs=output,
71
  examples=exemplos
72
  )
73
 
74
  # Executando a interface
75
+ demo.launch(share=False)