Merge branch 'main' into dependabot/npm_and_yarn/inference/landing/np…

…m_and_yarn-af93afb32e
roboflow · Dec 13, 2024 · a11a08b · a11a08b
2 parents a00a58e + 40d8277
commit a11a08b
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/inference/core/workflows/core_steps/models/foundation/google_gemini/v1.py b/inference/core/workflows/core_steps/models/foundation/google_gemini/v1.py
@@ -166,11 +166,11 @@ class BlockManifest(WorkflowBlockManifest):
     )
     model_version: Union[
         Selector(kind=[STRING_KIND]),
-        Literal["gemini-1.5-flash", "gemini-1.5-pro"],
+        Literal["gemini-2.0-flash-exp", "gemini-1.5-flash", "gemini-1.5-pro"],
     ] = Field(
         default="gemini-1.5-flash",
         description="Model to be used",
-        examples=["gemini-1.5-flash", "$inputs.gemini_model"],
+        examples=["gemini-2.0-flash-exp", "$inputs.gemini_model"],
     )
     max_tokens: int = Field(
         default=450,
@@ -543,7 +543,10 @@ def prepare_ocr_prompt(
                         "mime_type": "image/jpeg",
                         "data": base64_image,
                     }
-                }
+                },
+                {
+                    "text": f"Read the text",
+                },
             ],
             "role": "user",
         },
@@ -581,7 +584,10 @@ def prepare_caption_prompt(
                         "mime_type": "image/jpeg",
                         "data": base64_image,
                     }
-                }
+                },
+                {
+                    "text": f"Caption the image",
+                },
             ],
             "role": "user",
         },