Update __init__.py

aaravnavani · web-flow · commit 006e3c15945d · 2024-08-29T11:13:44.000-07:00
diff --git a/validator/__init__.py b/validator/__init__.py
@@ -1,3 +1,3 @@
-from .main import ValidatorTemplate
+from .main import ShieldGemma2B
 
-__all__ = ["ValidatorTemplate"]
+__all__ = ["ShieldGemma2B"]
