Update model_parsers.py

model_parsers.py

@@ -1,4 +1,5 @@
-from aiconfig import AIConfigRuntime, ModelParserRegistry, DalleImageGenerationParser
+from aiconfig import AIConfigRuntime, ModelParserRegistry #, DalleImageGenerationParser
+from aiconfig_extension_gemini import GeminiModelParser
 
 
 # Example of how users can register model parsers for use in the GradioWorkbook
@@ -7,11 +8,14 @@ from aiconfig import AIConfigRuntime, ModelParserRegistry, DalleImageGenerationP
 # Here we are registering all the local HuggingFace model parsers as an example
 def register_model_parsers() -> None:
 
-    dalle_image_generation_models = [
-        "dall-e-2",
-        "dall-e-3",
-    ]
-    for model in dalle_image_generation_models:
-        ModelParserRegistry.register_model_parser(
-            DalleImageGenerationParser(model)
-        )
+    # dalle_image_generation_models = [
+    #     "dall-e-2",
+    #     "dall-e-3",
+    # # ]
+    # for model in dalle_image_generation_models:
+    #      ModelParserRegistry.register_model_parser(
+    #         DalleImageGenerationParser(model)
+    #     )
+    ModelParserRegistry.register_model_parser(
+        GeminiModelParser("gemini-pro"), ["gemini-pro"]
+    )