Added Gemini to models init.py

10 months ago · 5062bf3e7a
parent dedf458a8a
commit 5062bf3e7a
1 changed files with 2 additions and 0 deletions
--- a/swarms/models/init.py
+++ b/swarms/models/init.py
@ -14,6 +14,7 @@ from swarms.models.nougat import Nougat  # noqa: E402
 from swarms.models.openai_embeddings import OpenAIEmbeddings
 from swarms.models.openai_tts import OpenAITTS  # noqa: E402
 from swarms.models.palm import GooglePalm as Palm  # noqa: E402
+from swarms.models.gemini import Gemini
 from swarms.models.popular_llms import Anthropic as Anthropic
 from swarms.models.popular_llms import (
    AzureOpenAILLM as AzureOpenAI,
@ -46,6 +47,7 @@ __all__ = [
    "BaseLLM",
    "BaseMultiModalModel",
    "Fuyu",
+    "Gemini",
    "GPT4VisionAPI",
    "HuggingfaceLLM",
    "Idefics",