atomkevich commited on
Commit
d9f0474
·
verified ·
1 Parent(s): eb750b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,7 +20,7 @@ translation_model = M2M100ForConditionalGeneration.from_pretrained(model_name)
20
  @tool
21
  def agent_translate(text: str, detected_lang: str) -> str:
22
  """A tool that translate text with next logic:
23
- 1) If text starts with 'PL:', translate the rest to Polish.
24
  2) If detected language == 'ru', translate to English.
25
  3) Else translate to Russian.
26
  and return translated text
@@ -28,7 +28,7 @@ def agent_translate(text: str, detected_lang: str) -> str:
28
  text: text in russian/english or polish languages.
29
  detected_lang: detected language of the text
30
  """
31
- if text.startswith("PL:"):
32
  original = text[3:].strip()
33
  target_lang = "PL"
34
  else:
 
20
  @tool
21
  def agent_translate(text: str, detected_lang: str) -> str:
22
  """A tool that translate text with next logic:
23
+ 1) If text starts with 'pl:', translate the rest to Polish.
24
  2) If detected language == 'ru', translate to English.
25
  3) Else translate to Russian.
26
  and return translated text
 
28
  text: text in russian/english or polish languages.
29
  detected_lang: detected language of the text
30
  """
31
+ if text.startswith("pl:"):
32
  original = text[3:].strip()
33
  target_lang = "PL"
34
  else: