ai-forever commited on
Commit
0a89e97
·
verified ·
1 Parent(s): 08a71c3

change googletrans to deep_translator

Browse files
Files changed (1) hide show
  1. src/llm_utils.py +6 -3
src/llm_utils.py CHANGED
@@ -1,13 +1,16 @@
1
  from typing import List, Callable
2
- from googletrans import Translator
 
3
  import random
4
 
5
  from src.prompt_configs import PromptConfig, prefix
6
 
7
- translator = Translator()
 
8
 
9
  def get_translation(text: str, dest: str = 'en') -> str:
10
- return translator.translate(text, dest=dest).text
 
11
 
12
  def llm_generate_titles(
13
  llm_generate: Callable[[str], str],
 
1
  from typing import List, Callable
2
+ # from googletrans import Translator
3
+ from deep_translator import GoogleTranslator
4
  import random
5
 
6
  from src.prompt_configs import PromptConfig, prefix
7
 
8
+ # translator = Translator()
9
+ translator = GoogleTranslator(source='auto', target='en')
10
 
11
  def get_translation(text: str, dest: str = 'en') -> str:
12
+ return translator.translate(text)
13
+ # return translator.translate(text, dest=dest).text
14
 
15
  def llm_generate_titles(
16
  llm_generate: Callable[[str], str],