diff --git a/g4f/client/image_models.py b/g4f/client/image_models.py
index 0c8ea894..1fd2d0b5 100644
--- a/g4f/client/image_models.py
+++ b/g4f/client/image_models.py
@@ -1,3 +1,9 @@
+from .Provider.BingCreateImages import BingCreateImages
+from .Provider.needs_auth import Gemini, OpenaiChat
+from ..Provider.You import You
+
+from .types import Client
+
 class ImageModels():
     gemini = Gemini
     openai = OpenaiChat