JeffersonCorreiax commited on
Commit
438c827
·
verified ·
1 Parent(s): 54a2946

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import torch
2
  from transformers import T5Tokenizer, T5ForConditionalGeneration
3
  import gradio as gr
4
- from googletrans import Translator
5
 
6
  # Inicialize o tokenizer e o modelo
7
  tokenizer = T5Tokenizer.from_pretrained('t5-small')
@@ -10,14 +10,11 @@ model = T5ForConditionalGeneration.from_pretrained('cssupport/t5-small-awesome-t
10
  model = model.to(device)
11
  model.eval()
12
 
13
- # Inicialize o tradutor
14
- translator = Translator()
15
-
16
  # Função para traduzir português para inglês
17
  def traduzir_para_ingles(texto):
18
  try:
19
- traducao = translator.translate(texto, src='pt', dest='en')
20
- return traducao.text
21
  except Exception as e:
22
  print(f"Erro na tradução: {e}")
23
  return texto # Retorna o texto original em caso de erro
 
1
  import torch
2
  from transformers import T5Tokenizer, T5ForConditionalGeneration
3
  import gradio as gr
4
+ from deep_translator import GoogleTranslator
5
 
6
  # Inicialize o tokenizer e o modelo
7
  tokenizer = T5Tokenizer.from_pretrained('t5-small')
 
10
  model = model.to(device)
11
  model.eval()
12
 
 
 
 
13
  # Função para traduzir português para inglês
14
  def traduzir_para_ingles(texto):
15
  try:
16
+ traducao = GoogleTranslator(source='pt', target='en').translate(texto)
17
+ return traducao
18
  except Exception as e:
19
  print(f"Erro na tradução: {e}")
20
  return texto # Retorna o texto original em caso de erro