Update ocr.py

torrmal · web-flow · commit ae9041dbae53 · 2024-10-09T12:41:44.000-04:00
no max tokens
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

Original file line number	Diff line number	Diff line change
`@@ -48,8 +48,7 @@ def process_image_to_markdown(file_object, client, model="gpt-4o", prompt = DEF`
`48`	`48`	`}`
`49`	`49`	`]`
`50`	`50`	`}`
`51`		`- ],`
`52`		`- max_tokens=1000`
	`51`	`+ ]`
`53`	`52`	`)`
`54`	`53`
`55`	`54`	`# Extract the markdown content from the response`