handle lowercase to uppercase

NexaAI · Nov 22, 2024 · c8e922f · c8e922f
1 parent da38f91
commit c8e922f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/nexa/gguf/nexa_inference_vlm_omni.py b/nexa/gguf/nexa_inference_vlm_omni.py
@@ -149,6 +149,9 @@ def run(self):
 
     def inference(self, prompt: str, image_path: str):
         with suppress_stdout_stderr():
+            if prompt and prompt[0].islower():
+                prompt = prompt[0].upper() + prompt[1:]
+
             prompt = ctypes.c_char_p(prompt.encode("utf-8"))
             image_path = ctypes.c_char_p(image_path.encode("utf-8"))
             response = omni_vlm_cpp.omnivlm_inference(prompt, image_path)