Adding singlish _translator_tool

app.py

@@ -55,7 +55,7 @@ with open("concised_prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, image_generation_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, image_generation_tool, singlish_translator_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,

tools/singlish_translator.py

@@ -0,0 +1,15 @@
+from smolagents import Tool
+from huggingface_hub import InferenceClient
+
+class SinglishTranslatorTool(Tool):
+    name = "singlish_translator"
+    description = "A tool that translates English to Singlish."
+    inputs = {'text': {'type': 'string', 'description': 'The text to translate.'}}
+    output_type = "string"
+    model = "h2oai/danube2-singlish-finetuned"
+    client = InferenceClient(model)
+    
+    def forward(self, text: str) -> str:
+        return self.client.generate(text)
+    
+