diff --git a/swarms/models/__init__.py b/swarms/models/__init__.py index f5975b5f..173c9487 100644 --- a/swarms/models/__init__.py +++ b/swarms/models/__init__.py @@ -31,7 +31,7 @@ from swarms.models.layoutlm_document_qa import ( ) # noqa: E402 from swarms.models.gpt4_vision_api import GPT4VisionAPI # noqa: E402 from swarms.models.openai_tts import OpenAITTS # noqa: E402 - +from swarms.models.gemini import Gemini # noqa: E402 # from swarms.models.gpt4v import GPT4Vision # from swarms.models.dalle3 import Dalle3 # from swarms.models.distilled_whisperx import DistilWhisperModel # noqa: E402 @@ -63,4 +63,5 @@ __all__ = [ "GPT4VisionAPI", # "vLLM", "OpenAITTS", + "Gemini", ]