janmoons commited on
Commit
da5d358
·
verified ·
1 Parent(s): cb259fc

add the provider and explicitly added inputs parameter

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -16,10 +16,11 @@ def translate_text(text:str, source_language:str, target_language:str="nl")-> st
16
  source_language: the language of the input text
17
  target_language: the language the input text needs to be translated to
18
  """
19
- client = InferenceClient()
20
  return client.translation(
21
- text,
22
  model="facebook/nllb-200-1.3B",
 
23
  src_lang=source_language,
24
  tgt_lang=target_language
25
  )
 
16
  source_language: the language of the input text
17
  target_language: the language the input text needs to be translated to
18
  """
19
+ client = InferenceClient(provider="hf-inference")
20
  return client.translation(
21
+ inputs=text,
22
  model="facebook/nllb-200-1.3B",
23
+ provider="hf-inference",
24
  src_lang=source_language,
25
  tgt_lang=target_language
26
  )