Create model_parsers.py

model_parsers.py

@@ -0,0 +1,56 @@
+from aiconfig_extension_hugging_face import (
+    # HuggingFaceAutomaticSpeechRecognitionTransformer,
+    # HuggingFaceImage2TextTransformer,
+    HuggingFaceText2ImageDiffusor,
+    # HuggingFaceText2SpeechTransformer,
+    # HuggingFaceTextGenerationTransformer,
+    # HuggingFaceTextSummarizationTransformer,
+    # HuggingFaceTextTranslationTransformer,
+)
+
+from aiconfig import AIConfigRuntime, ModelParserRegistry
+
+
+# Example of how to register model parsers for use in the GradioWorkbook
+# The implementation looks for a parsers_path (model_parsers.py by default) which
+# should include a module with a register_model_parsers function.
+# Here we are registering the local HuggingFace model parsers as an example
+def register_model_parsers() -> None:
+    """Register model parsers for local HuggingFace models."""
+    # automatic_speech_recognition = HuggingFaceAutomaticSpeechRecognitionTransformer()
+    # AIConfigRuntime.register_model_parser(
+    #     automatic_speech_recognition, "Automatic Speech Recognition (Local)"
+    # )
+
+    # image_to_text = HuggingFaceImage2TextTransformer()
+    # AIConfigRuntime.register_model_parser(image_to_text, "Image-to-Text (Local)")
+
+    text_to_image = HuggingFaceText2ImageDiffusor()
+    AIConfigRuntime.register_model_parser(text_to_image, "Text-to-Image (Local)")
+
+    # text_to_speech = HuggingFaceText2SpeechTransformer()
+    # AIConfigRuntime.register_model_parser(text_to_speech, "Text-to-Speech (Local)")
+
+    # text_generation = HuggingFaceTextGenerationTransformer()
+    # AIConfigRuntime.register_model_parser(text_generation, "Text Generation (Local)")
+
+    # text_summarization = HuggingFaceTextSummarizationTransformer()
+    # AIConfigRuntime.register_model_parser(text_summarization, "Summarization (Local)")
+
+    # text_translation = HuggingFaceTextTranslationTransformer()
+    # AIConfigRuntime.register_model_parser(text_translation, "Translation (Local)")
+
+    # By default, model parsers will also have their own ids registered. Remove those
+    # since we just want the task-based names registered
+    parsers = [
+        # automatic_speech_recognition,
+        # image_to_text,
+        text_to_image,
+        # text_to_speech,
+        # text_generation,
+        # text_summarization,
+        # text_translation,
+    ]
+
+    for parser in parsers:
+        ModelParserRegistry.remove_model_parser(parser.id())