Update ocr.py

no max tokens
mindsdb · Oct 9, 2024 · ae9041d · ae9041d
1 parent 3592504
commit ae9041d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/aipdf/ocr.py b/src/aipdf/ocr.py
@@ -48,8 +48,7 @@ def process_image_to_markdown(file_object, client, model="gpt-4o",  prompt = DEF
                         }
                     ]
                 }
-            ],
-            max_tokens=1000
+            ]
         )
 
         # Extract the markdown content from the response