sarath2803 commited on
Commit
02432b1
·
verified ·
1 Parent(s): 7e4b2e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -4,21 +4,21 @@ import requests
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
7
- from deep_translator import GoogleTranslator
8
 
9
  from Gradio_UI import GradioUI
10
 
11
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
12
  @tool
13
- def my_custom_tool(arg1:str, arg2:str, arg3:str)-> str: #it's import to specify the return type
14
  #Keep this format for the description / args / args description but feel free to modify the tool
15
  """A tool that translates a sentence from arg1 to arg2
16
  Args:
17
  arg1: the sentence
18
- arg2: the original language
19
- arg3: language to which arg1 is to be translated
20
  """
21
- result=GoogleTranslator(source=arg2,target=arg3).translate(arg1)
 
22
  return f"Translated sentence is : {result}"
23
 
24
  @tool
 
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
7
+ from translate import Translator
8
 
9
  from Gradio_UI import GradioUI
10
 
11
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
12
  @tool
13
+ def my_custom_tool(arg1:str, arg2:str)-> str: #it's import to specify the return type
14
  #Keep this format for the description / args / args description but feel free to modify the tool
15
  """A tool that translates a sentence from arg1 to arg2
16
  Args:
17
  arg1: the sentence
18
+ arg2: language to which arg1 is to be translated
 
19
  """
20
+ translator = Translator(to_lang=arg2)
21
+ result=translator.translate(arg1)
22
  return f"Translated sentence is : {result}"
23
 
24
  @tool