first commit

README.md

@@ -10,4 +10,10 @@ pinned: false
 license: apache-2.0
 ---
 
+Make sure to set you API key as an environment variable:
+```
+export REMYXAI_API_KEY=<your-key-here>
+```
+
+Read more: https://github.com/remyxai/remyxai-cli
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,4 @@
+from transformers.tools.base import launch_gradio_demo
+from classifier_labeler import RemyxClassifier
+
+launch_gradio_demo(RemyxClassifier)

classifier_labeler.py

@@ -0,0 +1,25 @@
+from transformers import Tool
+from remyxai.utils import labeler
+
+class RemyxClassifier(Tool):
+    name = "custom_classifier"
+    description = (
+        """
+    This is a tool that fetches or trains a custom classifier 
+    and returns a json file containing the labels of the images in a directory.
+    It takes an image directory and a comma separated string of labels
+    as input, and returns the json file containing the predicted labels 
+    for each image as output
+    """
+    )
+
+    inputs = ["text", "text"]
+    outputs = ["text"]
+
+    def __call__(self, image_directory: str, labels: str):
+        labels_list = [l.strip() for l in labels.split(",")]
+        model_name = "labeler_{}".format("_".join(labels_list))
+        result = labeler(labels=labels_list, 
+                         image_dir=image_directory, 
+                         model_name=model_name)
+        return result

requirements.txt

@@ -0,0 +1,3 @@
+remyxai
+huggingface_hub
+transformers

tool_config.json

@@ -0,0 +1,5 @@
+{
+  "description": "This is a tool that fetches or trains a custom classifier and returns a json file containing the labels of the images in a directory. It takes an image directory and a comma separated string of labels as input, and returns the json file containing the predicted labels for each image as output",
+  "name": "custom_classifier",
+  "tool_class": "classifier_labeler.RemyxClassifier"
+}