Update ocr_llm_utils.py

ocr_llm_utils.py

@@ -28,7 +28,7 @@ def run_ocr_with_gcv(image_path):
     response = client_vision.document_text_detection(image=image)
     return response.full_text_annotation.text
 
-def extract_table_from_text(text,max_tokens=4096):
+def extract_table_from_text(text,max_tokens=4096,model="meta-llama/llama-4-scout-17b-16e-instruct"):
     prompt = f"""
     Extract a structured table of items from the invoice text below. 
     - First findout what are the table column names
@@ -44,7 +44,7 @@ def extract_table_from_text(text,max_tokens=4096):
     \"\"\"
     """
     response = client.chat.completions.create(
-        model="meta-llama/llama-4-scout-17b-16e-instruct",
+        model=model,
     messages=[
         {"role": "system", "content": "You are a professional invoice data extractor."},
         {"role": "user", "content": prompt}