Spaces:
Sleeping
Sleeping
add the provider and explicitly added inputs parameter
Browse files
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 |
)
|