add pipeline tags

abhishekkrthakur · abhishekkrthakur · commit d137c8c3477a · 2025-01-21T08:53:23.000+01:00
diff --git a/src/autotrain/__init__.py b/src/autotrain/__init__.py
@@ -45,7 +45,7 @@
 warnings.filterwarnings("ignore", category=UserWarning, module="huggingface_hub")
 
 logger = Logger().get_logger()
-__version__ = "0.8.36.dev0"
+__version__ = "0.8.36"
 
 
 def is_colab():
diff --git a/src/autotrain/trainers/image_classification/utils.py b/src/autotrain/trainers/image_classification/utils.py
@@ -34,6 +34,7 @@
 ---
 tags:
 - autotrain
+- transformers
 - image-classification{base_model}
 widget:
 - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg
diff --git a/src/autotrain/trainers/image_regression/utils.py b/src/autotrain/trainers/image_regression/utils.py
@@ -18,6 +18,7 @@
 
 MODEL_CARD = """
 ---
+library_name: transformers
 tags:
 - autotrain
 - vision
diff --git a/src/autotrain/trainers/object_detection/utils.py b/src/autotrain/trainers/object_detection/utils.py
@@ -27,6 +27,7 @@
 
 MODEL_CARD = """
 ---
+library_name: transformers
 tags:
 - autotrain
 - object-detection
diff --git a/src/autotrain/trainers/seq2seq/utils.py b/src/autotrain/trainers/seq2seq/utils.py
@@ -9,6 +9,7 @@
 
 MODEL_CARD = """
 ---
+library_name: transformers
 tags:
 - autotrain
 - text2text-generation{base_model}
diff --git a/src/autotrain/trainers/text_classification/utils.py b/src/autotrain/trainers/text_classification/utils.py
@@ -30,6 +30,7 @@
 
 MODEL_CARD = """
 ---
+library_name: transformers
 tags:
 - autotrain
 - text-classification{base_model}
diff --git a/src/autotrain/trainers/token_classification/utils.py b/src/autotrain/trainers/token_classification/utils.py
@@ -6,6 +6,7 @@
 
 MODEL_CARD = """
 ---
+library_name: transformers
 tags:
 - autotrain
 - token-classification{base_model}
