diff --git a/__init__.py b/__init__.py
index fd304dd..5742203 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,11 +1,14 @@
 # __init__.py
 from .layered_infinite_zoom import LayeredInfiniteZoom
+from .replicate_API import APIGenerateReplicate
 
 NODE_CLASS_MAPPINGS = {
-    "LayeredInfiniteZoom": LayeredInfiniteZoom
+    "LayeredInfiniteZoom": LayeredInfiniteZoom,
+    "APIGenerateReplicate": APIGenerateReplicate
 }
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "LayeredInfiniteZoom": "Layered Infinite Zoom"
+    "LayeredInfiniteZoom": "Layered Infinite Zoom",
+    "APIGenerateReplicate": "Replicate API"
 }
 
 __all__ = [