Upload tool

aitool.py

@@ -0,0 +1,15 @@
+from transformers import Tool
+
+
+class AITranslationTool(Tool):
+    name = "model_translation"
+    description = (
+        "This is a tool that translates "
+    )
+
+    inputs = ["text"]
+    outputs = ["text"]
+
+    def __call__(self, task: str):
+        model = "facebook/nllb-200-distilled-600M"
+        return model.id

app.py

@@ -0,0 +1,4 @@
+from transformers import launch_gradio_demo
+from aitool import AITranslationTool
+
+launch_gradio_demo(AITranslationTool)

requirements.txt

@@ -0,0 +1 @@
+transformers

tool_config.json

@@ -0,0 +1,5 @@
+{
+  "description": "This is a tool that translates ",
+  "name": "model_translation",
+  "tool_class": "aitool.AITranslationTool"
+}