fixdev: add model_repo_id to translation using facebooks'
Browse files
app.py
CHANGED
@@ -71,7 +71,7 @@ custom_role_conversions=None,
|
|
71 |
|
72 |
# Import tool from Hub
|
73 |
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True) # https://huggingface.co/spaces/agents-course/text-to-image
|
74 |
-
translation_tool = load_tool("smolagents-tools/translation", trust_remote_code=True)
|
75 |
|
76 |
with open("prompts.yaml", 'r') as stream:
|
77 |
prompt_templates = yaml.safe_load(stream)
|
|
|
71 |
|
72 |
# Import tool from Hub
|
73 |
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True) # https://huggingface.co/spaces/agents-course/text-to-image
|
74 |
+
translation_tool = load_tool("smolagents-tools/translation", model_repo_id="facebook/nllb-200-distilled-600M", trust_remote_code=True)
|
75 |
|
76 |
with open("prompts.yaml", 'r') as stream:
|
77 |
prompt_templates = yaml.safe_load(stream)
|