Upload model/heads/__init__.py with huggingface_hub

model/heads/__init__.py

@@ -0,0 +1,16 @@
+"""
+`model.heads` package exports.
+
+This file exposes the commonly used head classes so callers can
+import them from `model.heads` directly, e.g.:
+
+	from model.heads import NERHead, QAHead, SentimentHead
+
+Keeping an explicit `__all__` makes the package's public API clear.
+"""
+
+from .ner_head import NERHead
+from .qa_head import QAHead
+from .sentiment_head import SentimentHead
+
+__all__ = ["NERHead", "QAHead", "SentimentHead"]