llm_ticket3/test_agent_ocr.py
2025-04-29 10:04:52 +02:00

11 lines
406 B
Python

from agents.llama_vision.agent_vision_ocr import AgentVisionOCR
from llm_classes.llama_vision import LlamaVision
model = LlamaVision()
agent = AgentVisionOCR(model)
images = ["output/ticket_T11143/T11143_20250422_084617/attachments/image_145435.png"]
for img in images:
res = agent.executer(img, ocr_baseline="", ticket_id="")
print(f"Résultat pour l'image {img}:\n{res['extracted_text']}\n\n")